Update app.py

app.py

@@ -2,10 +2,10 @@ import gradio as gr
 import transformers
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 
-shroom_generator = pipeline("text-generation", model=AutoModelForCausalLM.from_pretrained('LorenzoDeMattei/GePpeTto'), tokenizer="gpt2")
+shroom_generator = pipeline("text-generation", model=AutoModelForCausalLM.from_pretrained('LorenzoDeMattei/GePpeTto'), tokenizer='LorenzoDeMattei/GePpeTto')
 
 def generator(inizia_la_storia = ''):
-  shroom_result = shroom_generator(inizia_la_storia, max_length=420) 
+  shroom_result = shroom_generator(inizia_la_storia, max_length=120) 
   return shroom_result[0]["generated_text"]
 
 iface = gr.Interface(fn=generator, inputs="text", outputs="text")