Update app.py

app.py

@@ -8,7 +8,7 @@ tokenizer = GPT2Tokenizer.from_pretrained(
     'LorenzoDeMattei/GePpeTto',
 )
 
-shroom_generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pretrained('azaninello/GPT2-icc'),
+shroom_generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pretrained('LorenzoDeMattei/GePpeTto'),
     tokenizer='LorenzoDeMattei/GePpeTto', 
     do_sample=True,
     max_length=100,
@@ -17,7 +17,7 @@ shroom_generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pr
     repetition_penalty=9.5)
 
 def generator(inizia_la_storia = ''):
-  shroom_result = shroom_generator(inizia_la_storia, max_length=520) 
+  shroom_result = shroom_generator(inizia_la_storia, max_length=420) 
   return shroom_result[0]["generated_text"]
 
 iface = gr.Interface(fn=generator, inputs="text", outputs="text")