Update app.py

app.py

@@ -4,19 +4,20 @@ from transformers import pipeline
 
 pipe = pipeline('text-generation', framework='pt', model='akhooli/ap2023', tokenizer='akhooli/ap2023')
 #gc.collect()
-
+samples = [['أنت'],['لولا كتاب'],['ألا ليت'],['يا قدس'],['عيد بأية حال'],['لكل شيء إذا ما'],['.']]
 def sayPoetry(prompt):
     gen = pipe(prompt, max_length=96, temperature = 0.95)[0]["generated_text"]
     poetry =""
     for line in gen.split('.')[:-1]:
-        poetry += line
+        poetry += line + "\n"
     return poetry
                 
-tb = gr.Textbox(label="Enter your prompt or select from examples:", placeholder="أنا الذي", lines=1)
-poetry = gr.Interface(fn=sayPoetry, inputs=tb, outputs=["text"], allow_flagging='never',
+input_prompt = gr.inputs.Textbox(label="Enter your prompt or select from examples:", placeholder="أنا الذي", lines=1)
+output_text = gr.outputs.Textbox(label="Generated Poetry:", lines=4)
+poetry = gr.Interface(fn=sayPoetry, inputs=input_prompt, outputs=output_text, allow_flagging='never',
                    title='Arabic Poetry Generation Demo (updated Jan. 2023)',
                    description = "a simple demo of AI generated poetry based on 1M poems fine-tuned using AraGPT2 (be patient, runs on cpu)",
-                   examples=[['أنت'],['لولا كتاب'],['ألا ليت'],['يا قدس'],['عيد بأية حال'],['لكل شيء إذا ما'],['.']],
+                   examples=samples,
                      ,theme = 'darkhuggingface')
 
 poetry.launch()