Hugging Face's logo

Spaces:
BigSalmon
/
FormalInformalConciseWordy
Runtime error

App Files Community
BigSalmon commited on
Commit
738039e
·
1 Parent(s): 08112a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -25,10 +25,13 @@ 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/InformalToFormalLincoln63Paraphrase")
29
- model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln68Paraphrase")
30
- model2 = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln63Paraphrase")
31
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln63Paraphrase")
 
 
 
32
  return model, model2, tokenizer
33
 
34
  model, model2, 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/InformalToFormalLincoln63Paraphrase")
29
+ #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln68Paraphrase")
30
+ #model2 = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln63Paraphrase")
31
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln63Paraphrase")
32
+ tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln72Paraphrase")
33
+ model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln72Paraphrase")
34
+ model2 = AutoModelForCausalLM.from_pretrained("gpt2")
35
  return model, model2, tokenizer
36
 
37
  model, model2, tokenizer = get_model()