stanley-00 commited on
Commit
20ab2a9
·
verified ·
1 Parent(s): 0c3d719

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -133,11 +133,11 @@ with gr.Blocks(title="Small MF Model Tester", theme=gr.themes.Soft()) as app:
133
  with gr.Column(scale=1):
134
 
135
  with gr.Accordion("System Monitoring", open=True):
136
- stats_output = gr.Textbox(label="Live System Stats")
137
  gr.Timer(2).tick(get_system_stats, None, stats_output)
138
 
139
  with gr.Group():
140
- gr.Markdown("## Model Loader\nSelect or Paste custom model id here...")
141
  with gr.Row():
142
  model_id_input = gr.Dropdown(choices=MODELS, label="Model", allow_custom_value=True, show_label=False, scale=3)
143
  load_btn = gr.Button("Load", variant="secondary", scale=1)
@@ -147,7 +147,7 @@ with gr.Blocks(title="Small MF Model Tester", theme=gr.themes.Soft()) as app:
147
 
148
  with gr.Accordion("Generation Configuration", open=False):
149
  do_sample_input = gr.Checkbox(label="Enable Sampling (do_sample)", value=True, info="Uncheck for greedy decoding")
150
- max_tokens_input = gr.Slider(minimum=10, maximum=2048, value=128, step=1, label="Max Output Tokens")
151
  temperature_input = gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.1, label="Temperature", info="Higher = more creative")
152
 
153
  gr.Markdown("#### Advanced Sampling Constraints")
@@ -160,7 +160,7 @@ with gr.Blocks(title="Small MF Model Tester", theme=gr.themes.Soft()) as app:
160
  with gr.Column(scale=2):
161
  user_prompt = gr.Textbox(
162
  label="Prompt",
163
- value="Once upon a time in a digital kingdom,",
164
  placeholder="Enter your prompt here...",
165
  lines=5
166
  )
 
133
  with gr.Column(scale=1):
134
 
135
  with gr.Accordion("System Monitoring", open=True):
136
+ stats_output = gr.Textbox(label="Live System Stats", show_label=False)
137
  gr.Timer(2).tick(get_system_stats, None, stats_output)
138
 
139
  with gr.Group():
140
+ gr.Markdown("### Select or Paste custom model id here...")
141
  with gr.Row():
142
  model_id_input = gr.Dropdown(choices=MODELS, label="Model", allow_custom_value=True, show_label=False, scale=3)
143
  load_btn = gr.Button("Load", variant="secondary", scale=1)
 
147
 
148
  with gr.Accordion("Generation Configuration", open=False):
149
  do_sample_input = gr.Checkbox(label="Enable Sampling (do_sample)", value=True, info="Uncheck for greedy decoding")
150
+ max_tokens_input = gr.Slider(minimum=10, maximum=2048, value=256, step=1, label="Max Output Tokens")
151
  temperature_input = gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.1, label="Temperature", info="Higher = more creative")
152
 
153
  gr.Markdown("#### Advanced Sampling Constraints")
 
160
  with gr.Column(scale=2):
161
  user_prompt = gr.Textbox(
162
  label="Prompt",
163
+ value="Once upon a time in a digital kingdom, ",
164
  placeholder="Enter your prompt here...",
165
  lines=5
166
  )