Update app.py

app.py

@@ -53,7 +53,7 @@ def visualize_prediction(pil_img, output_dict, threshold=0.5, id2label=None):
     plt.axis("off")
     return fig2img(plt.gcf())
 
-def detect_objects(model_name,url_input,image_input,threshold):
+def detect_objects(model_name,url_input,image_input,webcam_input,threshold):
     
     #Extract model and feature extractor
     feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
@@ -65,6 +65,9 @@ def detect_objects(model_name,url_input,image_input,threshold):
         
     elif image_input:
         image = image_input
+        
+    elif webcam_input:
+        image = webcam_input
     
     #Make prediction
     processed_outputs = make_prediction(image, feature_extractor, model)
@@ -139,10 +142,17 @@ with demo:
                                                    
                 
             img_but = gr.Button('Detect')
-        
-    
-    url_but.click(detect_objects,inputs=[options,url_input,img_input,slider_input],outputs=img_output_from_url,queue=True)
-    img_but.click(detect_objects,inputs=[options,url_input,img_input,slider_input],outputs=img_output_from_upload,queue=True)
+            
+        with gr.TabItem('WebCam'):
+            with gr.Row():
+                web_input = gr.Image(source='webcam',type='pil',shape=(750,750))
+                img_output_from_webcam= gr.Image(shape=(750,750))
+
+            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)
     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])