Update app.py

app.py

@@ -1,22 +1,21 @@
 import torch
 from ultralytics import YOLO
 from ultralytics.nn.tasks import DetectionModel
+from ultralytics.nn.modules.conv import Conv
 import torch.nn as nn
 import gradio as gr
 
-# ✅ Whitelist required classes for PyTorch 2.6 safe loading
-torch.serialization.add_safe_globals([DetectionModel, nn.Sequential])
+# ✅ Whitelist all necessary classes for PyTorch 2.6 safe loading
+torch.serialization.add_safe_globals([DetectionModel, nn.Sequential, Conv])
 
 # Load YOLOv8 model
-model = YOLO("best.pt")
+model = YOLO("best (1).pt")  # Change to your model file path
 
 # Detection function
 def detect_fracture(image):
-    # Run detection
     results = model.predict(source=image, save=False)
-    annotated_img = results[0].plot()  # Annotated image with bounding boxes
+    annotated_img = results[0].plot()
 
-    # Extract labels & confidence scores
     detections = []
     for box in results[0].boxes:
         cls_id = int(box.cls[0])