Spaces:
Sleeping
Sleeping
Improve demo defaults
Browse files
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=
|
| 46 |
-
temperature = gr.Slider(0.2, 1.5, value=0.
|
| 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)
|