a

app.py

@@ -3,7 +3,7 @@ import cv2
 from ultralytics import YOLO
 
 # Load YOLO model
-model = YOLO("best.pt")
+model = YOLO('best.pt')
 
 # Function to perform object detection on an image
 def show_preds_image(image_path):
@@ -17,30 +17,14 @@ def show_preds_image(image_path):
             (int(det[2]), int(det[3])),
             color=(0, 0, 255),
             thickness=2,
-            lineType=cv2.LINE_AA,
+            lineType=cv2.LINE_AA
         )
     return cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
 
-
-# Define inputs and outputs for image interface
-inputs_image = [
-    gr.inputs.Image(type="filepath", label="Input Image"),
-]
-outputs_image = [
-    gr.outputs.Image(type="numpy", label="Output Image"),
-]
-interface_image = gr.Interface(
-    fn=show_preds_image,
-    inputs=inputs_image,
-    outputs=outputs_image,
-    title="Garbage Detector (Image Input)",
-)
-
-
 # Function to perform object detection on a video stream
 def show_preds_video(video_stream):
     cap = cv2.VideoCapture(video_stream.name)
-    while cap.isOpened():
+    while(cap.isOpened()):
         ret, frame = cap.read()
         if ret:
             frame_copy = frame.copy()
@@ -53,26 +37,32 @@ def show_preds_video(video_stream):
                     (int(det[2]), int(det[3])),
                     color=(0, 0, 255),
                     thickness=2,
-                    lineType=cv2.LINE_AA,
+                    lineType=cv2.LINE_AA
                 )
             yield cv2.cvtColor(frame_copy, cv2.COLOR_BGR2RGB)
 
+# Define interfaces for image and video inference
+interface_image = gr.Interface(
+    fn=show_preds_image,
+    inputs=gr.inputs.Image(type="file", label="Upload Image"),
+    outputs=gr.outputs.Image(type="numpy", label="Output Image"),
+    title="Pothole Detector (Image Input)",
+)
 
-# Define inputs and outputs for video interface
-inputs_video = [
-    gr.inputs.Video(type="webcam", label="Input Webcam"),
-]
-outputs_video = [
-    gr.outputs.Image(type="numpy", label="Output Image"),
-]
 interface_video = gr.Interface(
     fn=show_preds_video,
-    inputs=inputs_video,
-    outputs=outputs_video,
-    title="Garbage Detector (Webcam Input)",
+    inputs=gr.inputs.Video(type="webcam", label="Webcam Input"),
+    outputs=gr.outputs.Image(type="numpy", label="Output Image"),
+    title="Pothole Detector (Webcam Input)",
 )
 
 # Launch tabbed interface for both image and video inference
-gr.TabbedInterface(
-    [interface_image, interface_video], tab_names=["Image Inference", "Video Inference"]
+gr.Interface(
+    [interface_image, interface_video],
+    title="Pothole Detector",
+    description="Detect potholes using YOLOv8 on images and webcam streams.",
+    examples=[
+        ["path/to/image.jpg"],
+        ["path/to/video.mp4"]
+    ]
 ).launch()