Spaces:

BigSalmon
/

GPT2Space

Runtime error

BigSalmon commited on May 29, 2022

Commit

3f6088e

1 Parent(s): 48f445a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ first = """informal english: corn fields are all across illinois, visible once y
 @st.cache(allow_output_mutation=True)
 def get_model():
-    model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln46")
     tokenizer = AutoTokenizer.from_pretrained("BigSalmon/Points2")
     return model, tokenizer

 @st.cache(allow_output_mutation=True)
 def get_model():
+    model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln25")
     tokenizer = AutoTokenizer.from_pretrained("BigSalmon/Points2")
     return model, tokenizer