axay28 commited on
Commit
61d5788
·
verified ·
1 Parent(s): 981fac6

Improve demo defaults

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -42,8 +42,8 @@ with gr.Blocks(title="Tiny Transformer") as demo:
42
  with gr.Row():
43
  with gr.Column():
44
  prompt = gr.Textbox(value="To be", label="Prompt", lines=5)
45
- max_new_tokens = gr.Slider(8, 240, value=80, step=1, label="New tokens")
46
- temperature = gr.Slider(0.2, 1.5, value=0.8, step=0.05, label="Temperature")
47
  top_k = gr.Slider(1, 30, value=10, step=1, label="Top-k")
48
  button = gr.Button("Generate", variant="primary")
49
  output = gr.Textbox(label="Output", lines=16)
 
42
  with gr.Row():
43
  with gr.Column():
44
  prompt = gr.Textbox(value="To be", label="Prompt", lines=5)
45
+ max_new_tokens = gr.Slider(8, 240, value=120, step=1, label="New tokens")
46
+ temperature = gr.Slider(0.2, 1.5, value=0.55, step=0.05, label="Temperature")
47
  top_k = gr.Slider(1, 30, value=10, step=1, label="Top-k")
48
  button = gr.Button("Generate", variant="primary")
49
  output = gr.Textbox(label="Output", lines=16)