Update app.py

app.py

@@ -11,7 +11,7 @@ import gradio as gr
 torch.serialization.add_safe_globals([DetectionModel, nn.Sequential, Conv])
 
 # ---- Load trained YOLO model ----
-model = YOLO("best.pt")  # Ensure your model file is in the same folder
+model = YOLO("res.pt")  # Ensure your model file is in the same folder
 
 # ---- Prediction function ----
 def predict(image):