BigSalmon commited on
Commit
a909f4e
1 Parent(s): e29aa1d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,8 +25,8 @@ def get_model():
25
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPT2Neo1.3BPoints2")
26
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPT2Neo1.3BPoints3")
27
  #model = AutoModelForCausalLM.from_pretrained("facebook/opt-125m")
28
- tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln58Paraphrase")
29
- model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln58Paraphrase")
30
  return model, tokenizer
31
 
32
  model, tokenizer = get_model()
25
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPT2Neo1.3BPoints2")
26
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPT2Neo1.3BPoints3")
27
  #model = AutoModelForCausalLM.from_pretrained("facebook/opt-125m")
28
+ tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln62Paraphrase")
29
+ model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln62Paraphrase")
30
  return model, tokenizer
31
 
32
  model, tokenizer = get_model()