Update app.py

app.py

@@ -2,9 +2,11 @@ import gradio as gr
 import cv2
 import numpy as np
 from ultralytics import YOLO
+from huggingface_hub import hf_hub_download
 
 # Initialize models
-duck_model = YOLO('https://huggingface.co/brainwavecollective/yolo8n-rubber-duck-detector/resolve/main/yolov8n_rubberducks.pt')
+model_path = hf_hub_download(repo_id="brainwavecollective/yolov8n-rubber-duck-detector", filename="yolov8n_rubberducks.pt")
+duck_model = YOLO(model_path)
 standard_model = YOLO('yolov8n.pt')
 
 def process_image(image, model, is_standard_model=True):
@@ -63,7 +65,7 @@ def compare_models(input_image):
 
 # Create Gradio interface with stacked layout
 with gr.Blocks() as iface:
-    gr.Markdown("# Improved YOLO Rubber Duck Detection)
+    gr.Markdown("# Improved YOLO Rubber Duck Detection")
     gr.Markdown("Compare standard YOLOv8 model (left) with Fine-tuned model (right)")
     
     with gr.Column():