Update app.py

app.py

@@ -55,14 +55,22 @@ def predict_video(video):
 
 # Define Gradio interface
 iface = gr.Interface(
-    fn={"Image": predict_image, "Video": predict_video},
-    inputs={"Image": gr.inputs.Image(type="numpy", label="Upload an Image"),
-            "Video": gr.inputs.Video(label="Upload a Video")},
-    outputs={"Image": gr.outputs.Image(type="numpy", label="Detected Objects"),
-             "Video": gr.outputs.Video(label="Processed Video")},
+    fn=predict_image,
+    inputs=gr.Image(type="numpy", label="Upload an Image"),
+    outputs=gr.Image(type="numpy", label="Detected Objects"),
     title="YOLOv8 Object Detection",
-    description="Upload an image or a video and get detected objects using YOLOv8"
+    description="Upload an image and get detected objects using YOLOv8"
 )
 
+iface_video = gr.Interface(
+    fn=predict_video,
+    inputs=gr.Video(label="Upload a Video"),
+    outputs=gr.Video(label="Processed Video"),
+    title="YOLOv8 Object Detection",
+    description="Upload a video and get detected objects using YOLOv8"
+)
+
+demo = gr.TabbedInterface([iface, iface_video], ["Image Detection", "Video Detection"])
+
 if __name__ == "__main__":
-    iface.launch()
+    demo.launch()