Update app.py

app.py

@@ -2,13 +2,20 @@ import streamlit as st
 from ultralytics import YOLO
 from PIL import Image
 import numpy as np
+import os
 
-# Correct path to the model file in the root directory
-model_path = "best.pt"  # Ensure this matches the exact name of your model file
+# Define the model path
+model_path = "best.pt"  # Assumes `best.pt` is in the root directory
+
+# Verify file existence
+if not os.path.exists(model_path):
+    st.error(f"Model file not found at: {model_path}. Current directory files: {os.listdir()}")
+
+# Load the YOLO model
 try:
     model = YOLO(model_path)
-except FileNotFoundError:
-    st.error(f"Model file not found: {model_path}. Please ensure 'best.pt' is uploaded to the root directory.")
+except Exception as e:
+    st.error(f"Error loading model: {e}")
 
 # Streamlit app
 st.title("YOLOv11 Object Detection")
@@ -27,4 +34,4 @@ if uploaded_file:
             st.write("Detection Results:")
             st.image(results[0].plot(), caption="Detections", use_column_width=True)
         except Exception as e:
-            st.error(f"Error during prediction: {str(e)}")
+            st.error(f"E