BigSalmon commited on
Commit
f851efc
1 Parent(s): 05d7eef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -34,9 +34,10 @@ def get_model():
34
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln39")
35
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln40")
36
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln44")
 
37
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/Points4")
38
  tokenizer = AutoTokenizer.from_pretrained("BigSalmon/GPTNeo350MInformalToFormalLincoln")
39
- model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln46")
40
  return model, tokenizer
41
 
42
  model, tokenizer = get_model()
 
34
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln39")
35
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln40")
36
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln44")
37
+ #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln46")
38
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/Points4")
39
  tokenizer = AutoTokenizer.from_pretrained("BigSalmon/GPTNeo350MInformalToFormalLincoln")
40
+ model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln50")
41
  return model, tokenizer
42
 
43
  model, tokenizer = get_model()