Update app.py

app.py

@@ -13,7 +13,8 @@ shroom_generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pr
     do_sample=True,
     max_length=100,
     top_k=50,
-    top_p=0.95,)
+    top_p=0.95,
+    repetition_penalty=0.1)
 
 def generator(inizia_la_storia = ''):
   shroom_result = shroom_generator(inizia_la_storia, max_length=520)