Update app.py

app.py

@@ -7,7 +7,7 @@ import time
 
 # Load the YOLO model only once using Streamlit session state
 if 'models_loaded' not in st.session_state:
-    st.session_state.yolo_model = YOLO('/kaggle/working/classification_project/yolo_classification/weights/best.pt')  # Update with your model path
+    st.session_state.yolo_model = YOLO('best.pt')  # Update with your model path
     st.session_state.models_loaded = True
 
 # Define function for inference using YOLO