Update app.py

app.py

@@ -69,12 +69,7 @@ def detect_objects(model_name,url_input,image_input,webcam_input,threshold):
     
     if validators.url(url_input):
         image = get_original_image(url_input)
-        #Make prediction
-        processed_outputs = make_prediction(image, feature_extractor, model)
-        #Visualize prediction
-        viz_img = visualize_prediction(image, processed_outputs, threshold, model.config.id2label)
          
-        
     elif image_input:
         image = image_input
         
@@ -165,9 +160,9 @@ with demo:
 
             cam_but = gr.Button('Detect')
             
-    url_but.click(detect_objects,inputs=[options,url_input,img_input,web_input,slider_input],outputs=[img_output_from_url,original_image],queue=True)
-    img_but.click(detect_objects,inputs=[options,url_input,img_input,web_input,slider_input],outputs=[img_output_from_upload,original_image],queue=True)
-    cam_but.click(detect_objects,inputs=[options,url_input,img_input,web_input,slider_input],outputs=[img_output_from_webcam,original_image],queue=True)
+    url_but.click(detect_objects,inputs=[options,url_input,img_input,web_input,slider_input],outputs=[img_output_from_url],queue=True)
+    img_but.click(detect_objects,inputs=[options,url_input,img_input,web_input,slider_input],outputs=[img_output_from_upload],queue=True)
+    cam_but.click(detect_objects,inputs=[options,url_input,img_input,web_input,slider_input],outputs=[img_output_from_webcam],queue=True)
     example_images.click(fn=set_example_image,inputs=[example_images],outputs=[img_input])
     example_url.click(fn=set_example_url,inputs=[example_url],outputs=[url_input,original_image])