akhooli commited on
Commit
18c6822
1 Parent(s): 5eb8066

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -12
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, 50, 1.0, 1.0, 114],['هل غادر'
9
- ,1.0, 50, 1.0, 1.0, 114 ],['ألا ليت'
10
- ,1.0, 50, 1.0, 1.0, 114 ],['يا قدس'
11
- ,1.0, 50, 1.0, 1.0, 114],['عيد بأية حال'
12
- ,1.0, 50, 1.0, 1.0, 114],['لكل شيء إذا ما'
13
- ,1.0, 50, 1.0, 1.0, 114 ],['.'
14
- ,1.0, 50, 1.0, 1.0, 114]]
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.70, 1.2, step=0.01,value=1.0, label='control temperature'),
40
- gr.Slider(25, 100, step=1,value=50, label='control top k'),
41
- gr.Slider(0.80, 1.0, step=0.01,value=1.0, label='control top p'),
42
- gr.Slider(0.90, 1.50, step=0.01,value=1.0, label='control penalty'),
43
- gr.Number(value=139750, precision=0, label='Seed'),
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