a

.gitignore

@@ -1,5 +1,4 @@
 flagged/
-*.pt
 *.png
 *.jpg
 *.mp4

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,10 +17,11 @@ 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"),
@@ -35,10 +36,11 @@ interface_image = gr.Interface(
     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()
@@ -51,10 +53,11 @@ 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 inputs and outputs for video interface
 inputs_video = [
     gr.inputs.Video(type="webcam", label="Input Webcam"),
@@ -71,6 +74,5 @@ interface_video = gr.Interface(
 
 # Launch tabbed interface for both image and video inference
 gr.TabbedInterface(
-    [interface_image, interface_video],
-    tab_names=['Image Inference', 'Video Inference']
+    [interface_image, interface_video], tab_names=["Image Inference", "Video Inference"]
 ).launch()

best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2f688e18f683030746d48dacf9b60b198722205e3837afd5b2d26574f541a8f2
+size 22520163