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- 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=
|
498 |
-
noise_slider = gr.Slider(0, 50, value=
|
499 |
-
sharpen_slider = gr.Slider(0, 50, value=
|
500 |
-
confidence_slider = gr.Slider(0.0, 1.0, value=0.
|
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=
|
528 |
inputs=inputs,
|
529 |
outputs=outputs,
|
530 |
api_name="predict"
|
531 |
)
|
532 |
augment_button.click( # Connect Augment button to the function
|
533 |
-
fn=
|
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,
|