Update app.py

app.py

@@ -4,7 +4,14 @@ from transformers import pipeline
 
 pipe = pipeline('text-generation', framework='pt', model='akhooli/ap2023', tokenizer='akhooli/ap2023')
 #gc.collect()
-samples = [['أنت'],['هل غادر'],['ألا ليت'],['يا قدس'],['عيد بأية حال'],['لكل شيء إذا ما'],['.']]
+samples = [['أنت'
+           ,None, None, None, None],['هل غادر'
+            ,None, None, None, None ],['ألا ليت'
+              ,None, None, None, None ],['يا قدس'
+                 ,None, None, None, None],['عيد بأية حال'
+                   ,None, None, None, None],['لكل شيء إذا ما'
+                    ,None, None, None, None ],['.'
+                       ,None, None, None, None]]
 #samples = [['أنت'
 #           ,.96],['هل غادر'
 #             , 1.0],['ألا ليت'
@@ -30,11 +37,11 @@ def sayPoetry(prompt, temp=0.95, topk = 50, topp = 0.92, penalty=1.0):
     return poetry
 poetry = gr.Interface(fn=sayPoetry, 
                     inputs=[
-                        gr.inputs.Textbox(label="Enter short prompt or select from examples:", placeholder="أنا الذي"), 
-                        gr.inputs.Slider(0.80, 1.0, step=0.01,default=0.95, label='control  temperature'),
-                        gr.inputs.Slider(25, 100, step=1,default=50, label='control  top k'),
-                        gr.inputs.Slider(0.80, 1.0, step=0.01,default=1.0, label='control  top p'),
-                        gr.inputs.Slider(0.90, 1.20, step=0.01,default=1.20, label='control  penalty'),
+                        gr.Textbox(label="Enter short prompt or select from examples:", placeholder="أنا الذي"), 
+                        gr.Slider(0.80, 1.0, step=0.01,default=0.95, label='control  temperature'),
+                        gr.Slider(25, 100, step=1,default=50, label='control  top k'),
+                        gr.Slider(0.80, 1.0, step=0.01,default=1.0, label='control  top p'),
+                        gr.Slider(0.90, 1.20, step=0.01,default=1.20, label='control  penalty'),
                     ], 
                 outputs=[gr.outputs.Textbox(label="Generated Poetry:")], 
                     
@@ -42,6 +49,6 @@ poetry = gr.Interface(fn=sayPoetry,
                 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,
-                cache_examples=False,
+                #cache_examples=False,
                 article = notes)
 poetry.launch()