Upload app.py

app.py

@@ -0,0 +1,39 @@
+import gradio as gr
+import torch
+import os
+
+# --- Configuration ---
+MODEL_REPO_ID = "Nawinkumar15/my-yolov8-model"  # Replace with YOUR model repository ID
+MODEL_FILENAME = "best.pt"
+CACHE_DIR = "model_cache"
+
+# Ensure cache directory exists
+os.makedirs(CACHE_DIR, exist_ok=True)
+
+# Function to load the YOLOv8 model from Hugging Face Hub
+def load_model():
+    try:
+        model = torch.hub.load('ultralytics/yolov8', 'custom', path=f'{CACHE_DIR}/{MODEL_FILENAME}', trust_repo=True)
+        return model
+    except Exception as e:
+        print(f"Error loading model: {e}")
+        return None
+
+# Load the model when the app starts
+model = load_model()
+
+def predict(image):
+    if model is None:
+        return "Model not loaded."
+    results = model(image)
+    return results[0].plot()
+
+iface = gr.Interface(
+    fn=predict,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Image(type="pil", label="Detected Objects"),
+    title="Roboflow YOLOv8 Detector",
+    description="Upload an image to detect objects using your YOLOv8 model trained with Roboflow."
+)
+
+iface.launch()