Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -69,10 +69,10 @@ with gr.Blocks(title="Small Model Tester") as app:
|
|
| 69 |
gr.Timer(5).tick(get_system_stats, None, stats_output)
|
| 70 |
|
| 71 |
with gr.Row():
|
| 72 |
-
model_id_input = gr.Textbox(
|
| 73 |
load_btn = gr.Button("Load", variant="secondary")
|
| 74 |
|
| 75 |
-
max_tokens_input = gr.Slider(minimum=10, maximum=1024, value=
|
| 76 |
temperature_input = gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.1, label="Temperature")
|
| 77 |
top_k_input = gr.Slider(minimum=1, maximum=100, value=50, step=1, label="Top-K Sampling")
|
| 78 |
|
|
|
|
| 69 |
gr.Timer(5).tick(get_system_stats, None, stats_output)
|
| 70 |
|
| 71 |
with gr.Row():
|
| 72 |
+
model_id_input = gr.Textbox(value="AxiomicLabs/GPT-S-5M", placeholder="e.g., AxiomicLabs/GPT-S-5M")
|
| 73 |
load_btn = gr.Button("Load", variant="secondary")
|
| 74 |
|
| 75 |
+
max_tokens_input = gr.Slider(minimum=10, maximum=1024, value=128, step=1, label="Max Output Tokens")
|
| 76 |
temperature_input = gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.1, label="Temperature")
|
| 77 |
top_k_input = gr.Slider(minimum=1, maximum=100, value=50, step=1, label="Top-K Sampling")
|
| 78 |
|