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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -29,8 +29,9 @@ 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
  tokenizer = AutoTokenizer.from_pretrained("BigSalmon/GPTNeo350MInformalToFormalLincoln")
33
- model = AutoModelForCausalLM.from_pretrained("BigSalmon/Points4")
34
  return model, tokenizer
35
 
36
  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/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()