Update app.py

app.py

@@ -2,7 +2,11 @@ import gradio as gr
 import transformers
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 
-shroom_generator = pipeline("text-generation", model=AutoModelForCausalLM.from_pretrained('LorenzoDeMattei/GePpeTto'), tokenizer='LorenzoDeMattei/GePpeTto')
+shroom_generator = pipeline("text-generation", model=AutoModelForCausalLM.from_pretrained('LorenzoDeMattei/GePpeTto'),
+    tokenizer='LorenzoDeMattei/GePpeTto', 
+    max_length=50,
+    top_k=50,
+    top_p=0.95,)
 
 def generator(inizia_la_storia = ''):
   shroom_result = shroom_generator(inizia_la_storia, max_length=120)