Update app.py

app.py

@@ -32,13 +32,16 @@ def run_inference(image):
 # Create the Gradio interface
 interface = gr.Interface(
     fn=run_inference,
-    inputs=gr.Image(type="pil"),
+    inputs=[
+        gr.Image(type="pil", label="Upload Image"),  # For file uploads
+        gr.Camera(type="pil", label="Capture from Camera")  # For camera capture
+    ],
     outputs=[
         gr.Image(type="pil"),
         gr.Textbox(label="Object Counts", lines=5, interactive=False)  # Display counts as text
     ],
     title="YOLOv5 Object Detection with Counts",
-    description="Upload an image to run YOLOv5 object detection, see the annotated results, and view the count of detected objects by category."
+    description="Upload an image or capture a photo to run YOLOv5 object detection, see the annotated results, and view the count of detected objects by category."
 )
 
 # Launch the app