Update app.py

app.py

@@ -3,9 +3,12 @@ from ultralytics import YOLO
 from PIL import Image
 import numpy as np
 
-# Load the YOLO model directly from the root directory
+# Correct path to the model file in the root directory
 model_path = "best.pt"  # Ensure this matches the exact name of your model file
-model = YOLO(model_path)
+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.")
 
 # Streamlit app
 st.title("YOLOv11 Object Detection")
@@ -19,8 +22,9 @@ if uploaded_file:
 
     # Perform prediction
     with st.spinner("Processing..."):
-        results = model.predict(np.array(image))
-    
-    # Display results
-    st.write("Detection Results:")
-    st.image(results[0].plot(), caption="Detections", use_column_width=True)
+        try:
+            results = model.predict(np.array(image))
+            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)}")