Update app.py

app.py

@@ -4,13 +4,15 @@ from transformers import pipeline
 
 pipe = pipeline('text-generation', framework='pt', model='akhooli/ap2023', tokenizer='akhooli/ap2023')
 #gc.collect()
-samples = [['أنت'
-           ,.96],['هل غادر'
+samples = [['أنت'],['هل غادر'],['ألا ليت'],['يا قدس'],['عيد بأية حال'],['لكل شيء إذا ما'],['.']]
+#samples = [['أنت'
+#           ,.96],['هل غادر'
              , 1.0],['ألا ليت'
-                    ,0.99],['يا قدس'
-                           , 0.98],['عيد بأية حال'
-                                   ,0.94],['لكل شيء إذا ما'
-                                          ,0.92],['.',0.99]]
+#                    ,0.99],['يا قدس'
+#                           , 0.98],['عيد بأية حال'
+#                                   ,0.94],['لكل شيء إذا ما'
+#                                          ,0.92],['.',0.99]]
+                     
 notes = """
 - Enter a short prompt or select (click) one of the examples and click SEND 
 - Adjust parameters (temperture, top k, top p and penalty) through the slider (keep close to default values).
@@ -31,7 +33,7 @@ poetry = gr.Interface(fn=sayPoetry,
                         gr.inputs.Slider(0.80, 1.0, step=0.01,default=0.95, label='control  temperature'),
                         gr.inputs.Slider(20, 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.89, 1.0, step=0.01,default=1.20, label='control  penalty'),
+                        gr.inputs.Slider(0.90, 1.20, step=0.01,default=1.20, label='control  penalty'),
                     ], 
                     outputs=[gr.outputs.Textbox(label="Generated Poetry:")],