nastasiasnk commited on
Commit
d3c7e21
1 Parent(s): d1117d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ HfFolder.save_token(HF_TOKEN)
10
  import transformers
11
  from transformers import pipeline
12
 
13
- generator = pipeline(model="openai-community/gpt2")
14
 
15
 
16
  text = st.text_area("your input")
 
10
  import transformers
11
  from transformers import pipeline
12
 
13
+ generator = pipeline(model="EleutherAI/gpt-neo-2.7B")
14
 
15
 
16
  text = st.text_area("your input")