Update app.py

app.py

@@ -75,13 +75,13 @@ def detect_objects(model_name,url_input,image_input,webcam_input,threshold):
     #Visualize prediction
     viz_img = visualize_prediction(image, processed_outputs, threshold, model.config.id2label)
     
-    return viz_img   
+    return viz_img, image   
         
 def set_example_image(example: list) -> dict:
     return gr.Image.update(value=example[0])
 
 def set_example_url(example: list) -> dict:
-    return gr.Textbox.update(value=example[0])
+    return gr.Textbox.update(value=example[0]), display_url_input.update(image)
 
 
 title = """<h1 id="title">Face Mask Detection with YOLOS</h1>"""
@@ -124,6 +124,7 @@ with demo:
         with gr.TabItem('Image URL'):
             with gr.Row():
                 url_input = gr.Textbox(lines=2,label='Enter valid image URL here..')
+                orig_image = gr.Image(shape=(750,750))
                 img_output_from_url = gr.Image(shape=(750,750))
                 
             with gr.Row():
@@ -150,9 +151,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,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)
+    url_but.click(detect_objects,inputs=[options,url_input,img_input,web_input,slider_input],outputs=[img_output_from_url,orig_image],queue=True)
+    img_but.click(detect_objects,inputs=[options,url_input,img_input,web_input,slider_input],outputs=[img_output_from_upload,orig_image],queue=True)
+    cam_but.click(detect_objects,inputs=[options,url_input,img_input,web_input,slider_input],outputs=[img_output_from_webcam,orig_image],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])