Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,13 +5,13 @@ from transformers import pipeline, set_seed
|
|
| 5 |
pipe = pipeline('text-generation', framework='pt', model='akhooli/ap2023', tokenizer='akhooli/ap2023')
|
| 6 |
#gc.collect()
|
| 7 |
samples = [['أنت'
|
| 8 |
-
,1.0,
|
| 9 |
-
,1.0,
|
| 10 |
-
,1.0,
|
| 11 |
-
,1.0,
|
| 12 |
-
,1.0,
|
| 13 |
-
,1.0,
|
| 14 |
-
,1.0,
|
| 15 |
|
| 16 |
notes = """
|
| 17 |
- Enter a short prompt or select (click) one of the examples and click SEND
|
|
@@ -36,11 +36,11 @@ def sayPoetry(prompt, temp=1.0, topk = 50, topp = 1.0, penalty=1.0, seed=114):
|
|
| 36 |
poetry = gr.Interface(fn=sayPoetry,
|
| 37 |
inputs=[
|
| 38 |
gr.Textbox(label="Enter short prompt or select from examples:"),
|
| 39 |
-
gr.Slider(0.
|
| 40 |
-
gr.Slider(
|
| 41 |
-
gr.Slider(0.
|
| 42 |
-
gr.Slider(0.
|
| 43 |
-
gr.Number(value=
|
| 44 |
],
|
| 45 |
outputs=[gr.Textbox(label="Generated Poetry:")],
|
| 46 |
|
|
|
|
| 5 |
pipe = pipeline('text-generation', framework='pt', model='akhooli/ap2023', tokenizer='akhooli/ap2023')
|
| 6 |
#gc.collect()
|
| 7 |
samples = [['أنت'
|
| 8 |
+
,1.0, 25, 0.8, 1.0, 114],['هل غادر'
|
| 9 |
+
,1.0, 25, 0.8, 1.0, 114 ],['ألا ليت'
|
| 10 |
+
,1.0, 25, 0.8, 1.0, 114 ],['يا قدس'
|
| 11 |
+
,1.0, 25, 0.8, 1.0, 114],['عيد بأية حال'
|
| 12 |
+
,1.0, 25, 0.8, 1.0, 114],['لكل شيء إذا ما'
|
| 13 |
+
,1.0, 25, 0.8, 1.0, 114 ],['.'
|
| 14 |
+
,1.0, 25, 0.8, 1.0, 114]]
|
| 15 |
|
| 16 |
notes = """
|
| 17 |
- Enter a short prompt or select (click) one of the examples and click SEND
|
|
|
|
| 36 |
poetry = gr.Interface(fn=sayPoetry,
|
| 37 |
inputs=[
|
| 38 |
gr.Textbox(label="Enter short prompt or select from examples:"),
|
| 39 |
+
gr.Slider(0.50, 1.5, step=0.01,value=1.0, label='control temperature'),
|
| 40 |
+
gr.Slider(5, 50, step=1,value=25, label='control top k'),
|
| 41 |
+
gr.Slider(0.70, 0.9, step=0.01,value=0.8, label='control top p'),
|
| 42 |
+
gr.Slider(0.10, 1.0, step=0.01,value=1.0, label='control penalty'),
|
| 43 |
+
gr.Number(value=1359719, precision=0, label='Seed'),
|
| 44 |
],
|
| 45 |
outputs=[gr.Textbox(label="Generated Poetry:")],
|
| 46 |
|