Update app.py

app.py

@@ -4,20 +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"]
+samples = [['أنت', 0.95],['لولا كتاب'],['ألا ليت'],['يا قدس'],['عيد بأية حال'],['لكل شيء إذا ما'],['.']]
+def sayPoetry(prompt, temp):
+    gen = pipe(prompt, max_length=96, temperature = temp)[0]["generated_text"]
     poetry =""
     for line in gen.split('.')[:-1]:
         poetry += line + "\n"
     return poetry
-                
+tempreature = gr.Slider(0.70, 1.0, value=0.94, step=0.01)],                
 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',
+poetry = gr.Interface(fn=sayPoetry, inputs=[input_prompt, temperature], 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=samples,
-                     ,theme = 'darkhuggingface')
+                   theme = 'darkhuggingface')
 
 poetry.launch()