Update app.py

app.py

@@ -4,19 +4,17 @@ 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")
+model_path = os.path.join(model_dir, "Yolov8-fintuned-on-potholes.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
+    annotated_img = results[0].plot()  
     return Image.fromarray(annotated_img)
 
-# Gradio Interface
+
 demo = gr.Interface(
     fn=detect_potholes,
     inputs=gr.Image(type="pil"),