Update app.py

app.py

@@ -2,10 +2,9 @@ import streamlit as st
 from ultralytics import YOLO
 from PIL import Image
 import numpy as np
-import os
 
-# Load the model
-model_path = "./models/best.pt"
+# Load the YOLO model directly from the root directory
+model_path = "best.pt"  # Ensure this matches the exact name of your model file
 model = YOLO(model_path)
 
 # Streamlit app