Update yolo.py

yolo.py

@@ -124,7 +124,7 @@ class FractureDetector:
 
 def detect_fractures_in_image(image):
     """Function to handle the detection and return results."""
-    model_path = 'yolov8n_custom.pkl'  # Model path on Hugging Face Spaces
+    model_path = 'yolov8n_custom_exported.pt'  # Model path on Hugging Face Spaces
     output_folder = '/app/output_images'  # Output folder structure
     detector = FractureDetector(model_path, output_folder)
     results = detector.detect_fractures(image)