Update app.py

app.py

@@ -26,37 +26,30 @@ import torch
 model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')
 
 # Define the face detector function
-def detect_faces(image):
+def detect_image(image):
     results = model(image)
     return results.render()[0]
 
 # Create Gradio interfaces for different modes
 webcam_interface = gr.Interface(
-    fn=detect_faces,
-    inputs=gr.inputs.Image(source="webcam", tool=None),
+    fn=detect_image,
+    inputs=gr.inputs.Image(source="upload"),
     outputs="image",
-    title="Webcam Mode"
+    title="Image"
 )
 
 file_interface = gr.Interface(
-    fn=detect_faces,
-    inputs=gr.inputs.Image(source="file"),
-    outputs="image",
-    title="File Mode"
-)
-
-url_interface = gr.Interface(
-    fn=detect_faces,
-    inputs=gr.inputs.Image(source="url"),
-    outputs="image",
-    title="URL Mode"
+    fn=detect_image,
+    inputs=gr.inputs.Video(source="upload"),
+    outputs="video",
+    title="Videoe"
 )
 
 # Create a list of interfaces
 interfaces = [webcam_interface, file_interface, url_interface]
 
 # Create the tabbed interface
-tabbed_interface = gr.Interface(interfaces, layout="horizontal")
+tabbed_interface = gr.TabbedInterface(interfaces, ["Image", "Video"])
 
 # Launch the tabbed interface
 tabbed_interface.launch(debug=True)