Simon Salmon commited on
Commit
397c96d
·
1 Parent(s): 03d1761

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ informal english:''')
30
 
31
  from transformers import AutoTokenizer, AutoModelWithLMHead
32
  tokenizer = AutoTokenizer.from_pretrained("gpt2")
33
- model = AutoModelWithLMHead.from_pretrained("BigSalmon/MrLincoln4")
34
 
35
  with st.form(key='my_form'):
36
  prompt = st.text_area(label='Enter sentence')
 
30
 
31
  from transformers import AutoTokenizer, AutoModelWithLMHead
32
  tokenizer = AutoTokenizer.from_pretrained("gpt2")
33
+ model = AutoModelWithLMHead.from_pretrained("BigSalmon/MrLincoln5")
34
 
35
  with st.form(key='my_form'):
36
  prompt = st.text_area(label='Enter sentence')