Spaces:

BigSalmon
/

FormalInformalConciseWordy

Runtime error

BigSalmon commited on May 23, 2022

Commit

b3fda2d

1 Parent(s): f35654c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -23,8 +23,8 @@ def get_model():
     #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPT2Neo1.3BPoints2")
     #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPT2Neo1.3BPoints3")
     #model = AutoModelForCausalLM.from_pretrained("facebook/opt-125m")
-    model = AutoModelForCausalLM.from_pretrained("facebook/opt-1.3b")
-    tokenizer = AutoTokenizer.from_pretrained("facebook/opt-1.3b")
     return model, tokenizer
 model, tokenizer = get_model()

     #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPT2Neo1.3BPoints2")
     #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPT2Neo1.3BPoints3")
     #model = AutoModelForCausalLM.from_pretrained("facebook/opt-125m")
+    model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln46")
+    tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln46")
     return model, tokenizer
 model, tokenizer = get_model()