a

app.py

@@ -6,28 +6,28 @@ import numpy as np
 
 from ultralytics import YOLO
 
-file_urls = [
-    'https://www.dropbox.com/s/b5g97xo901zb3ds/pothole_example.jpg?dl=1',
-    'https://www.dropbox.com/s/86uxlxxlm1iaexa/pothole_screenshot.png?dl=1',
-    'https://www.dropbox.com/s/7sjfwncffg8xej2/video_7.mp4?dl=1'
-]
+# file_urls = [
+#     'https://www.dropbox.com/s/b5g97xo901zb3ds/Garbage_example.jpg?dl=1',
+#     'https://www.dropbox.com/s/86uxlxxlm1iaexa/Garbage_screenshot.png?dl=1',
+#     'https://www.dropbox.com/s/7sjfwncffg8xej2/video_7.mp4?dl=1'
+# ]
 
-def download_file(url, save_name):
-    if not os.path.exists(save_name):
-        file = requests.get(url)
-        open(save_name, 'wb').write(file.content)
+# def download_file(url, save_name):
+#     if not os.path.exists(save_name):
+#         file = requests.get(url)
+#         open(save_name, 'wb').write(file.content)
 
-for i, url in enumerate(file_urls):
-    if 'mp4' in file_urls[i]:
-        download_file(
-            file_urls[i],
-            f"video.mp4"
-        )
-    else:
-        download_file(
-            file_urls[i],
-            f"image_{i}.jpg"
-        )
+# for i, url in enumerate(file_urls):
+#     if 'mp4' in file_urls[i]:
+#         download_file(
+#             file_urls[i],
+#             f"video.mp4"
+#         )
+#     else:
+#         download_file(
+#             file_urls[i],
+#             f"image_{i}.jpg"
+#         )
 
 model = YOLO('best.pt')
 path  = [['1.jpeg'], ['2.jpeg']]
@@ -90,24 +90,24 @@ interface_image = gr.Interface(
     fn=show_preds_image,
     inputs=inputs_image,
     outputs=outputs_image,
-    title="Pothole detector",
+    title="Garbage Detection",
     examples=path,
     cache_examples=False,
 )
 
 inputs_video = gr.Video(label="Input Video")
-outputs_video = gr.Image(label="Output Image")
+outputs_video = gr.Video(label="Output Image")
 interface_video = gr.Interface(
     fn=show_preds_video,
     inputs=inputs_video,
     outputs=outputs_video,
-    title="Pothole detector",
+    title="Garbage Detection",
     examples=video_path,
     cache_examples=False,
 )
 
-inputs_webcam = gr.Image(sources="webcam", streaming=True)
-outputs_webcam = gr.Image(label="Output Image")
+inputs_webcam = gr.Video(sources="webcam", streaming=True)
+outputs_webcam = gr.Video(label="Output Image")
 interface_webcam = gr.Interface(
     fn=show_preds_webcam,
     inputs=inputs_webcam,