LPX commited on
Commit
1260077
·
1 Parent(s): 4b48107

refactor: rename predict_image_with_json to predict_with_ensemble and update its implementation to return consensus label

Browse files
Files changed (1) hide show
  1. app_mcp.py +6 -6
app_mcp.py CHANGED
@@ -494,10 +494,10 @@ with gr.Blocks(css="#post-gallery { overflow: hidden !important;} .grid-wrap{ ov
494
  image_input = gr.Image(label="Upload Image to Analyze", sources=['upload', 'webcam'], type='pil')
495
  with gr.Accordion("Settings (Optional)", open=False, elem_id="settings_accordion"):
496
  augment_checkboxgroup = gr.CheckboxGroup(["rotate", "add_noise", "sharpen"], label="Augmentation Methods")
497
- rotate_slider = gr.Slider(0, 45, value=2, step=1, label="Rotate Degrees", visible=False)
498
- noise_slider = gr.Slider(0, 50, value=4, step=1, label="Noise Level", visible=False)
499
- sharpen_slider = gr.Slider(0, 50, value=11, step=1, label="Sharpen Strength", visible=False)
500
- confidence_slider = gr.Slider(0.0, 1.0, value=0.75, step=0.05, label="Confidence Threshold")
501
  inputs = [image_input, confidence_slider, augment_checkboxgroup, rotate_slider, noise_slider, sharpen_slider]
502
  predict_button = gr.Button("Predict")
503
  augment_button = gr.Button("Augment & Predict")
@@ -524,13 +524,13 @@ with gr.Blocks(css="#post-gallery { overflow: hidden !important;} .grid-wrap{ ov
524
  augment_checkboxgroup.change(lambda methods: gr.update(visible="sharpen" in methods), inputs=[augment_checkboxgroup], outputs=[sharpen_slider])
525
 
526
  predict_button.click(
527
- fn=predict_image_with_json,
528
  inputs=inputs,
529
  outputs=outputs,
530
  api_name="predict"
531
  )
532
  augment_button.click( # Connect Augment button to the function
533
- fn=predict_image_with_json,
534
  inputs=[
535
  image_input,
536
  confidence_slider,
 
494
  image_input = gr.Image(label="Upload Image to Analyze", sources=['upload', 'webcam'], type='pil')
495
  with gr.Accordion("Settings (Optional)", open=False, elem_id="settings_accordion"):
496
  augment_checkboxgroup = gr.CheckboxGroup(["rotate", "add_noise", "sharpen"], label="Augmentation Methods")
497
+ rotate_slider = gr.Slider(0, 45, value=0, step=1, label="Rotate Degrees", visible=False)
498
+ noise_slider = gr.Slider(0, 50, value=0, step=1, label="Noise Level", visible=False)
499
+ sharpen_slider = gr.Slider(0, 50, value=0, step=1, label="Sharpen Strength", visible=False)
500
+ confidence_slider = gr.Slider(0.0, 1.0, value=0.7, step=0.05, label="Confidence Threshold")
501
  inputs = [image_input, confidence_slider, augment_checkboxgroup, rotate_slider, noise_slider, sharpen_slider]
502
  predict_button = gr.Button("Predict")
503
  augment_button = gr.Button("Augment & Predict")
 
524
  augment_checkboxgroup.change(lambda methods: gr.update(visible="sharpen" in methods), inputs=[augment_checkboxgroup], outputs=[sharpen_slider])
525
 
526
  predict_button.click(
527
+ fn=predict_with_ensemble,
528
  inputs=inputs,
529
  outputs=outputs,
530
  api_name="predict"
531
  )
532
  augment_button.click( # Connect Augment button to the function
533
+ fn=predict_with_ensemble,
534
  inputs=[
535
  image_input,
536
  confidence_slider,