Spaces:

BigSalmon
/

GPT2_Most_Probable

Runtime error

BigSalmon commited on Mar 23, 2022

Commit

f0b81b5

1 Parent(s): 4d5c512

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 @st.cache(allow_output_mutation=True)
 def get_model():
     tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln21")
-    model = AutoModelWithLMHead.from_pretrained("BigSalmon/InformalToFormalLincoln21")
     return model, tokenizer
 model, tokenizer = get_model()

 @st.cache(allow_output_mutation=True)
 def get_model():
     tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln21")
+    model = AutoModelWithLMHead.from_pretrained("BigSalmon/MASKInformalToFormalLincoln30")
     return model, tokenizer
 model, tokenizer = get_model()