BigSalmon commited on
Commit
b84a1b7
·
1 Parent(s): d80c037

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -34,8 +34,10 @@ def get_model():
34
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln35")
35
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPT2Neo1.3BPoints")
36
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/Points4")
37
- tokenizer = AutoTokenizer.from_pretrained("BigSalmon/GPT2Neo1.3BPoints")
38
- model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPT2Neo1.3BPoints3")
 
 
39
  return model, tokenizer
40
 
41
  model, tokenizer = get_model()
 
34
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln35")
35
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPT2Neo1.3BPoints")
36
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/Points4")
37
+ #tokenizer = AutoTokenizer.from_pretrained("BigSalmon/GPT2Neo1.3BPoints")
38
+ #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPT2Neo1.3BPoints3")
39
+ tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln91Paraphrase")
40
+ model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln91Paraphrase")
41
  return model, tokenizer
42
 
43
  model, tokenizer = get_model()