Update app.py

app.py

@@ -1,5 +1,36 @@
-import streamlit as st
+import glob
+import gradio as gr
+from ultralytics import YOLO
+
+model_path = "best.pt"
+model = YOLO(model_path)
+
+
+PREDICT_KWARGS = {
+    "conf": 0.15,
+}
+
+
+def run(image_path):
+    results = model.predict(image_path, **PREDICT_KWARGS)
+    return results[0].plot()[:, :, ::-1]  # reverse channels for gradio
+
+
+title = "OOI RCA Digital Still Camera Benthic Megafauna 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()
 
-# Create a slider and display the squared value
-x = st.slider('Select a value', 0, 100)
-st.write(x, 'squared is', x * x)