Upload 4 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+best.torchscript filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,34 @@
+# -*- coding: utf-8 -*-
+"""app.ipynb
+
+Automatically generated by Colab.
+
+Original file is located at
+    https://colab.research.google.com/drive/1nIR_ak2SXq9fwQmKpq4eupLwQU_SI4fn
+"""
+
+import gradio as gr
+import torch
+import cv2
+from ultralytics import YOLO
+import numpy as np
+
+model = YOLO("best.torchscript")
+
+def predict(image):
+    results = model(image)
+    annotated = results[0].plot()
+    annotated_rgb = cv2.cvtColor(annotated, cv2.COLOR_BGR2RGB)
+    return annotated_rgb
+
+iface = gr.Interface(
+    fn=predict,
+    inputs=gr.Image(type="numpy", label="Upload an image"),
+    outputs=gr.Image(type="numpy", label="Detected image"),
+    title="YOLOv8 Object Detection",
+    description="Upload an image to detect objects using a YOLOv8 model in TorchScript format.",
+    examples=None
+)
+
+if __name__ == "__main__":
+    iface.launch()

best.torchscript

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7e3c0f08c1e1ba08508c3e929c8aba1bc16c066adab69db62c7db217c499edae
+size 13054046

requirements.txt

@@ -0,0 +1,9 @@
+altair
+pandas
+streamlit
+ultralytics
+gradio
+wandb
+opencv-python-headless
+torch
+numpy