Update app.py
Browse files
app.py
CHANGED
|
@@ -289,8 +289,8 @@ with gr.Blocks(css=custom_css) as demo:
|
|
| 289 |
cam_slider = gr.Slider(minimum=0.00, maximum=1.00, step=0.01, value=0.40, label="CAM Threshold", elem_classes="inferno-slider")
|
| 290 |
alpha_slider = gr.Slider(minimum=0.00, maximum=1.00, step=0.01, value=0.60, label="CAM Alpha")
|
| 291 |
with gr.Column():
|
| 292 |
-
tag_string = gr.Textbox(label="Tag String")
|
| 293 |
threshold_slider = gr.Slider(minimum=0.00, maximum=1.00, step=0.01, value=0.20, label="Tag Threshold")
|
|
|
|
| 294 |
label_box = gr.Label(label="Tag Predictions", num_top_classes=250, show_label=False)
|
| 295 |
|
| 296 |
gr.Markdown("""
|
|
@@ -307,7 +307,8 @@ with gr.Blocks(css=custom_css) as demo:
|
|
| 307 |
fn=run_classifier,
|
| 308 |
inputs=[image, threshold_slider],
|
| 309 |
outputs=[tag_string, label_box, original_image_state, sorted_tag_score_state],
|
| 310 |
-
show_progress='minimal'
|
|
|
|
| 311 |
)
|
| 312 |
|
| 313 |
image.clear(
|
|
@@ -320,28 +321,32 @@ with gr.Blocks(css=custom_css) as demo:
|
|
| 320 |
fn=create_tags,
|
| 321 |
inputs=[threshold_slider, sorted_tag_score_state],
|
| 322 |
outputs=[tag_string, label_box],
|
| 323 |
-
show_progress='hidden'
|
|
|
|
| 324 |
)
|
| 325 |
|
| 326 |
label_box.select(
|
| 327 |
fn=cam_inference,
|
| 328 |
inputs=[original_image_state, cam_slider, alpha_slider],
|
| 329 |
outputs=[image, cam_state],
|
| 330 |
-
show_progress='minimal'
|
|
|
|
| 331 |
)
|
| 332 |
|
| 333 |
cam_slider.input(
|
| 334 |
fn=create_cam_visualization_pil,
|
| 335 |
inputs=[original_image_state, cam_state, alpha_slider, cam_slider],
|
| 336 |
outputs=[image],
|
| 337 |
-
show_progress='hidden'
|
|
|
|
| 338 |
)
|
| 339 |
|
| 340 |
alpha_slider.input(
|
| 341 |
fn=create_cam_visualization_pil,
|
| 342 |
inputs=[original_image_state, cam_state, alpha_slider, cam_slider],
|
| 343 |
outputs=[image],
|
| 344 |
-
show_progress='hidden'
|
|
|
|
| 345 |
)
|
| 346 |
|
| 347 |
if __name__ == "__main__":
|
|
|
|
| 289 |
cam_slider = gr.Slider(minimum=0.00, maximum=1.00, step=0.01, value=0.40, label="CAM Threshold", elem_classes="inferno-slider")
|
| 290 |
alpha_slider = gr.Slider(minimum=0.00, maximum=1.00, step=0.01, value=0.60, label="CAM Alpha")
|
| 291 |
with gr.Column():
|
|
|
|
| 292 |
threshold_slider = gr.Slider(minimum=0.00, maximum=1.00, step=0.01, value=0.20, label="Tag Threshold")
|
| 293 |
+
tag_string = gr.Textbox(label="Tag String")
|
| 294 |
label_box = gr.Label(label="Tag Predictions", num_top_classes=250, show_label=False)
|
| 295 |
|
| 296 |
gr.Markdown("""
|
|
|
|
| 307 |
fn=run_classifier,
|
| 308 |
inputs=[image, threshold_slider],
|
| 309 |
outputs=[tag_string, label_box, original_image_state, sorted_tag_score_state],
|
| 310 |
+
show_progress='minimal',
|
| 311 |
+
show_progress_on=image
|
| 312 |
)
|
| 313 |
|
| 314 |
image.clear(
|
|
|
|
| 321 |
fn=create_tags,
|
| 322 |
inputs=[threshold_slider, sorted_tag_score_state],
|
| 323 |
outputs=[tag_string, label_box],
|
| 324 |
+
show_progress='hidden',
|
| 325 |
+
show_progress_on=image
|
| 326 |
)
|
| 327 |
|
| 328 |
label_box.select(
|
| 329 |
fn=cam_inference,
|
| 330 |
inputs=[original_image_state, cam_slider, alpha_slider],
|
| 331 |
outputs=[image, cam_state],
|
| 332 |
+
show_progress='minimal',
|
| 333 |
+
show_progress_on=image
|
| 334 |
)
|
| 335 |
|
| 336 |
cam_slider.input(
|
| 337 |
fn=create_cam_visualization_pil,
|
| 338 |
inputs=[original_image_state, cam_state, alpha_slider, cam_slider],
|
| 339 |
outputs=[image],
|
| 340 |
+
show_progress='hidden',
|
| 341 |
+
show_progress_on=image
|
| 342 |
)
|
| 343 |
|
| 344 |
alpha_slider.input(
|
| 345 |
fn=create_cam_visualization_pil,
|
| 346 |
inputs=[original_image_state, cam_state, alpha_slider, cam_slider],
|
| 347 |
outputs=[image],
|
| 348 |
+
show_progress='hidden',
|
| 349 |
+
show_progress_on=image
|
| 350 |
)
|
| 351 |
|
| 352 |
if __name__ == "__main__":
|