main app file updated to relocate detect and classify methods to seperate modules

app.py

@@ -1,30 +1,10 @@
-import os
 import gradio as gr
-import numpy as np
-import supervision as sv
-from ultralytics import YOLO
-
-
-os.system("wget https://media.roboflow.com/notebooks/examples/dog.jpeg")
-os.system("wget https://media.roboflow.com/notebooks/examples/dog-2.jpeg")
-
-
-def detect(image, weights):
-    model = YOLO(weights + '.pt')
-    result = model(image, verbose=False)[0]
-    detections = sv.Detections.from_ultralytics(result)
-    box_annotator = sv.BoxAnnotator()
-    annotated_image = box_annotator.annotate(image.copy(), detections=detections)
-
-    return annotated_image
+from classify_utils import classify_tab
+from detect_utils import detect_tab
 
 
 if __name__ == '__main__':
-    gr.Interface(
-        detect, [gr.Image(type="numpy"), gr.Dropdown(choices=["yolov5s", "yolov8s"])],
-        gr.Image(type="numpy"),
-        title="supervision",
-        examples=[["dog.jpeg", "yolov5s"], ["dog-2.jpeg", "yolov8s"]],
-        description=
-        "Gradio based demo for <a href='https://github.com/roboflow/supervision' style='text-decoration: underline' target='_blank'>roboflow/supervision</a>, We write your reusable computer vision tools."
-    ).launch()
+    gr.TabbedInterface(
+        [detect_tab, classify_tab],
+        tab_names=['Image Detection', 'Image Classification']
+    ).queue().launch()