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

minor edits

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -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.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
 
 
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='temperature (higher is more creative)'),
40
+ gr.Slider(5, 60, step=1,value=40, label='top k sampling'),
41
+ gr.Slider(0.70, 0.95, step=0.01,value=0.90, label='top p probabilities'),
42
+ gr.Slider(0.10, 1.5, step=0.01,value=1.0, label='repetition penalty'),
43
+ gr.Number(value=1359719, precision=0, label='random seed'),
44
  ],
45
  outputs=[gr.Textbox(label="Generated Poetry:")],
46