BigSalmon commited on
Commit
cd484c1
·
1 Parent(s): 6d9f740

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -28,8 +28,9 @@ def get_model():
28
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln30")
29
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln31")
30
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln32")
 
31
  tokenizer = AutoTokenizer.from_pretrained("BigSalmon/GPTNeo350MInformalToFormalLincoln")
32
- model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln33")
33
  return model, tokenizer
34
 
35
  model, tokenizer = get_model()
 
28
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln30")
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()