Update app.py

app.py

@@ -3,17 +3,30 @@ from ultralytics import YOLO
 from PIL import Image
 import numpy as np
 import os
+import requests
 
-# Define the model path
-model_path = "best.pt"  # Assumes `best.pt` is in the root directory
+# Google Drive file ID for best.pt
+FILE_ID = "1VE4I-_OMoC-wzzo-udzhJ896-WdFwJjU"
+FILE_URL = f"https://drive.google.com/uc?id={FILE_ID}"
+MODEL_PATH = "best.pt"
 
-# 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()}")
+# Download the model if not already present
+if not os.path.exists(MODEL_PATH):
+    st.info("Downloading model file from Google Drive...")
+    try:
+        response = requests.get(FILE_URL, stream=True)
+        response.raise_for_status()
+        with open(MODEL_PATH, "wb") as f:
+            for chunk in response.iter_content(chunk_size=8192):
+                if chunk:
+                    f.write(chunk)
+        st.success("Model file downloaded successfully!")
+    except Exception as e:
+        st.error(f"Error downloading model: {e}")
 
 # Load the YOLO model
 try:
-    model = YOLO(model_path)
+    model = YOLO(MODEL_PATH)
 except Exception as e:
     st.error(f"Error loading model: {e}")
 
@@ -34,4 +47,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"E
+            st.error(f"Error during prediction: {e}")