Create app.py

app.py

@@ -0,0 +1,28 @@
+import gradio as gr
+from PIL import Image
+from ultralytics import YOLO
+from huggingface_hub import snapshot_download
+import os
+
+# Download the model from HF Hub (if not already cached)
+model_dir = snapshot_download("cazzz307/Pothole-Finetuned-YoloV8")
+model_path = os.path.join(model_dir, "best.pt")
+
+# Load model
+model = YOLO(model_path)
+
+def detect_potholes(image: Image.Image):
+    results = model(image)
+    annotated_img = results[0].plot()  # Returns numpy array with drawings
+    return Image.fromarray(annotated_img)
+
+# Gradio Interface
+demo = gr.Interface(
+    fn=detect_potholes,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Image(type="pil"),
+    title="YOLOv8 Pothole Detection",
+    description="Upload a road image. The model will highlight detected potholes using a YOLOv8 model fine-tuned on pothole datasets."
+)
+
+demo.launch()