stanley-00 commited on
Commit
4f50497
·
verified ·
1 Parent(s): 41aa821

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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(label="Model ID", 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=50, step=10, 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
 
 
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