Update app.py

app.py

@@ -10,11 +10,13 @@ HfFolder.save_token(HF_TOKEN)
 import transformers
 from transformers import pipeline
 
-generator = pipeline("text-generation")
+generator = pipeline(model="openai-community/gpt2")
+
+
 text = st.text_area("your input")
 
 if text:
-    out = generator(text)
+    out = generator(text, do_sample=False)
     st.json(out)