Initial HF space

.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
+trash_mbari_09072023_640imgsz_50epochs_yolov8.pt filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,13 +1,11 @@
 ---
 title: Trash Detector
-emoji: 🚀
-colorFrom: red
-colorTo: red
+emoji: 🗑️
+colorFrom: blue
+colorTo: gray
 sdk: gradio
-sdk_version: 3.43.2
+sdk_version: 3.40.1
 app_file: app.py
 pinned: false
 license: cc-by-4.0
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,35 @@
+import glob
+import gradio as gr
+from ultralytics import YOLO
+
+model_path = "trash_mbari_09072023_640imgsz_50epochs_yolov8.pt"
+model = YOLO(model_path)
+
+
+PREDICT_KWARGS = {
+    "classes": 0,
+}
+
+
+def run(image_path):
+    results = model.predict(image_path, **PREDICT_KWARGS)
+    return results[0].plot()[:, :, ::-1]  # reverse channels for gradio
+
+
+title = "Trash Detector"
+description = (
+    ""
+)
+
+examples = glob.glob("images/*.png")
+
+interface = gr.Interface(
+    run,
+    inputs=[gr.components.Image(type="filepath")],
+    outputs=gr.components.Image(type="numpy"),
+    title=title,
+    description=description,
+    examples=examples,
+)
+
+interface.queue().launch()

requirements.txt

@@ -0,0 +1 @@
+ultralytics==8.0.158

trash_mbari_09072023_640imgsz_50epochs_yolov8.pt

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