Spaces:

BigSalmon
/

GPT2_Most_Probable

Runtime error

BigSalmon commited on Nov 16, 2022

Commit

4e7c469

1 Parent(s): a37ef4f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

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