gatesla commited on
Commit
3c77ab7
1 Parent(s): 039cf73

Update app.py

Browse files

Trying to make sure I can add to a new text box.

Files changed (1) hide show
  1. app.py +6 -3
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