Spaces:

BigSalmon
/

FormalInformalConciseWordy

Runtime error

BigSalmon commited on Nov 16, 2022

Commit

6cb6628

1 Parent(s): 38db286

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -43,8 +43,8 @@ def get_model():
     #tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln83Paraphrase")
     model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln87Paraphrase")
     tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln87Paraphrase")
-    tokenizer2 = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln90ParaphraseMedium")
-    model2 = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln90ParaphraseMedium")
     return model, model2, tokenizer, tokenizer2
 model, model2, tokenizer, tokenizer2 = get_model()

     #tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln83Paraphrase")
     model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln87Paraphrase")
     tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln87Paraphrase")
+    tokenizer2 = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincolnMedium")
+    model2 = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincolnMedium")
     return model, model2, tokenizer, tokenizer2
 model, model2, tokenizer, tokenizer2 = get_model()