CorneliusWang commited on
Commit
ff6c480
Β·
verified Β·
1 Parent(s): 4bbb666

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -224,7 +224,7 @@ with gr.Blocks(title="MentorFlow - Strategy Comparison") as demo:
224
  gpu_status = gr.Textbox(label="GPU Status", value=check_gpu(), interactive=False)
225
  refresh_btn = gr.Button("πŸ”„ Refresh GPU Status")
226
 
227
- refresh_btn.click(fn=check_gpu, outputs=gpu_status)
228
 
229
  # Parameters
230
  with gr.Row():
@@ -281,7 +281,8 @@ with gr.Blocks(title="MentorFlow - Strategy Comparison") as demo:
281
  run_btn.click(
282
  fn=run_comparison,
283
  inputs=[iterations, seed, use_deterministic, device],
284
- outputs=[output_text, output_plot]
 
285
  )
286
 
287
  gr.Markdown("""
 
224
  gpu_status = gr.Textbox(label="GPU Status", value=check_gpu(), interactive=False)
225
  refresh_btn = gr.Button("πŸ”„ Refresh GPU Status")
226
 
227
+ refresh_btn.click(fn=check_gpu, outputs=gpu_status, api_name="check_gpu")
228
 
229
  # Parameters
230
  with gr.Row():
 
281
  run_btn.click(
282
  fn=run_comparison,
283
  inputs=[iterations, seed, use_deterministic, device],
284
+ outputs=[output_text, output_plot],
285
+ api_name="run_comparison"
286
  )
287
 
288
  gr.Markdown("""