[update]edit main
Browse files
main.py
CHANGED
@@ -235,27 +235,27 @@ def main():
|
|
235 |
with gr.Column():
|
236 |
img_output_from_url = gr.Image()
|
237 |
|
|
|
|
|
238 |
with gr.Row():
|
239 |
gr.Examples(examples=example_urls,
|
240 |
inputs=[url_input, model_name, threshold_slider, iou_threshold_slider],
|
241 |
examples_per_page=5,
|
242 |
)
|
243 |
|
244 |
-
url_but = gr.Button("Detect")
|
245 |
-
|
246 |
with gr.TabItem("Image Upload"):
|
247 |
with gr.Row():
|
248 |
img_input = gr.Image(type="pil")
|
249 |
img_output_from_upload = gr.Image()
|
250 |
|
|
|
|
|
251 |
with gr.Row():
|
252 |
gr.Examples(examples=example_images,
|
253 |
inputs=[img_input, model_name, threshold_slider, iou_threshold_slider],
|
254 |
examples_per_page=5,
|
255 |
)
|
256 |
|
257 |
-
img_but = gr.Button("Detect")
|
258 |
-
|
259 |
inputs = [url_input, img_input, model_name, threshold_slider, iou_threshold_slider, classes_to_detect]
|
260 |
url_but.click(detr_object_detection, inputs=inputs, outputs=[img_output_from_url], queue=True)
|
261 |
img_but.click(detr_object_detection, inputs=inputs, outputs=[img_output_from_upload], queue=True)
|
|
|
235 |
with gr.Column():
|
236 |
img_output_from_url = gr.Image()
|
237 |
|
238 |
+
url_but = gr.Button("Detect")
|
239 |
+
|
240 |
with gr.Row():
|
241 |
gr.Examples(examples=example_urls,
|
242 |
inputs=[url_input, model_name, threshold_slider, iou_threshold_slider],
|
243 |
examples_per_page=5,
|
244 |
)
|
245 |
|
|
|
|
|
246 |
with gr.TabItem("Image Upload"):
|
247 |
with gr.Row():
|
248 |
img_input = gr.Image(type="pil")
|
249 |
img_output_from_upload = gr.Image()
|
250 |
|
251 |
+
img_but = gr.Button("Detect")
|
252 |
+
|
253 |
with gr.Row():
|
254 |
gr.Examples(examples=example_images,
|
255 |
inputs=[img_input, model_name, threshold_slider, iou_threshold_slider],
|
256 |
examples_per_page=5,
|
257 |
)
|
258 |
|
|
|
|
|
259 |
inputs = [url_input, img_input, model_name, threshold_slider, iou_threshold_slider, classes_to_detect]
|
260 |
url_but.click(detr_object_detection, inputs=inputs, outputs=[img_output_from_url], queue=True)
|
261 |
img_but.click(detr_object_detection, inputs=inputs, outputs=[img_output_from_upload], queue=True)
|