Update app.py
Browse filesTrying to make sure I can add to a new text box.
app.py
CHANGED
@@ -81,7 +81,8 @@ def detect_objects(model_name,url_input,image_input,threshold):
|
|
81 |
#Visualize prediction
|
82 |
viz_img = visualize_prediction(image, processed_outputs, threshold, model.config.id2label)
|
83 |
|
84 |
-
return viz_img, processed_outputs
|
|
|
85 |
|
86 |
def set_example_image(example: list) -> dict:
|
87 |
return gr.Image.update(value=example[0])
|
@@ -150,8 +151,10 @@ with demo:
|
|
150 |
img_but = gr.Button('Detect')
|
151 |
|
152 |
|
153 |
-
url_but.click(detect_objects,inputs=[options,url_input,img_input,slider_input],outputs=[img_output_from_url, output_text1],queue=True)
|
154 |
-
img_but.click(detect_objects,inputs=[options,url_input,img_input,slider_input],outputs=[img_output_from_upload, output_text2],queue=True)
|
|
|
|
|
155 |
example_images.click(fn=set_example_image,inputs=[example_images],outputs=[img_input])
|
156 |
example_url.click(fn=set_example_url,inputs=[example_url],outputs=[url_input])
|
157 |
|
|
|
81 |
#Visualize prediction
|
82 |
viz_img = visualize_prediction(image, processed_outputs, threshold, model.config.id2label)
|
83 |
|
84 |
+
# return [viz_img, processed_outputs]
|
85 |
+
return viz_img
|
86 |
|
87 |
def set_example_image(example: list) -> dict:
|
88 |
return gr.Image.update(value=example[0])
|
|
|
151 |
img_but = gr.Button('Detect')
|
152 |
|
153 |
|
154 |
+
# url_but.click(detect_objects,inputs=[options,url_input,img_input,slider_input],outputs=[img_output_from_url, output_text1],queue=True)
|
155 |
+
# img_but.click(detect_objects,inputs=[options,url_input,img_input,slider_input],outputs=[img_output_from_upload, output_text2],queue=True)
|
156 |
+
url_but.click(detect_objects,inputs=[options,url_input,img_input,slider_input],outputs=img_output_from_url,queue=True)
|
157 |
+
img_but.click(detect_objects,inputs=[options,url_input,img_input,slider_input],outputs=img_output_from_upload,queue=True)
|
158 |
example_images.click(fn=set_example_image,inputs=[example_images],outputs=[img_input])
|
159 |
example_url.click(fn=set_example_url,inputs=[example_url],outputs=[url_input])
|
160 |
|