Spaces:

BigSalmon
/

GPT2_Most_Probable

Runtime error

BigSalmon commited on Nov 29, 2021

Commit

0addd1d

1 Parent(s): e7172c5

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")
 from transformers import AutoTokenizer, AutoModelWithLMHead
 tokenizer = AutoTokenizer.from_pretrained("gpt2")
-model = AutoModelWithLMHead.from_pretrained("BigSalmon/MrLincoln8")
 with st.form(key='my_form'):
     prompt = st.text_area(label='Enter sentence')

 from transformers import AutoTokenizer, AutoModelWithLMHead
 tokenizer = AutoTokenizer.from_pretrained("gpt2")
+model = AutoModelWithLMHead.from_pretrained("BigSalmon/MrLincoln10")
 with st.form(key='my_form'):
     prompt = st.text_area(label='Enter sentence')