BigSalmon commited on
Commit
ec1293e
·
1 Parent(s): 0a9e414

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -29,9 +29,10 @@ def get_model():
29
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln31")
30
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln32")
31
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln33")
 
32
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/Points4")
33
  tokenizer = AutoTokenizer.from_pretrained("BigSalmon/GPTNeo350MInformalToFormalLincoln")
34
- model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln34")
35
  return model, tokenizer
36
 
37
  model, tokenizer = get_model()
 
29
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln31")
30
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln32")
31
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln33")
32
+ #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln34")
33
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/Points4")
34
  tokenizer = AutoTokenizer.from_pretrained("BigSalmon/GPTNeo350MInformalToFormalLincoln")
35
+ model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln35")
36
  return model, tokenizer
37
 
38
  model, tokenizer = get_model()