Update app.py

app.py

@@ -4,9 +4,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 
 shroom_generator = pipeline("text-generation", model=AutoModelForCausalLM.from_pretrained('azaninello/gpt2-finetuned-shrooms'), tokenizer="gpt2")
 
-def generator(start_your_text='Some day '):
-  shroom_result = shroom_generator(start_your_text, max_length=420) 
-  #general_result = general_generator(input_text, max_length=420) 
+def generator(scrivi_un_titolo = ""):
+  shroom_result = shroom_generator(f"<|HEADLINE|>{scrivi_un_titolo}<|TEXT|>", max_length=420) 
   return shroom_result[0]["generated_text"]
 
 iface = gr.Interface(fn=generator, inputs="text", outputs="text")