Update Google Drive file ID

app.py

@@ -1,22 +1,40 @@
 import sys
 from pathlib import Path
-
 import os
 import gdown
+import streamlit as st
 
 MODEL_PATH = "models/saved/final_model.pt"
-GOOGLE_DRIVE_URL = "https://drive.google.com/drive/folders/1VEFa0y_vW6AzT5x0fRwmX8shoBhUGd7K"  # Replace with your file's ID
+# You need to replace this with the direct download link of your model file
+# To get the direct link: Right-click your model file in Google Drive -> Get link -> Make sure it's set to "Anyone with the link can view"
+# Then replace the file ID in the URL below
+GOOGLE_DRIVE_FILE_ID = "1xhYKuC5_Yri3mm3Ejt-SpB2WAVUTJvc_"  # Replace with your actual file ID
+GOOGLE_DRIVE_URL = f"https://drive.google.com/uc?id={GOOGLE_DRIVE_FILE_ID}"
 
-if not os.path.exists(MODEL_PATH):
-    os.makedirs(os.path.dirname(MODEL_PATH), exist_ok=True)
-    gdown.download(GOOGLE_DRIVE_URL, MODEL_PATH, quiet=False)
+@st.cache_resource
+def download_model():
+    """Download model from Google Drive if not exists."""
+    if not os.path.exists(MODEL_PATH):
+        os.makedirs(os.path.dirname(MODEL_PATH), exist_ok=True)
+        with st.spinner("Downloading model from Google Drive..."):
+            try:
+                gdown.download(GOOGLE_DRIVE_URL, MODEL_PATH, quiet=False)
+                st.success("Model downloaded successfully!")
+            except Exception as e:
+                st.error(f"Failed to download model: {str(e)}")
+                st.error("Please check your Google Drive link and make sure the file is publicly accessible.")
+                return False
+    return True
 
 # Add src directory to Python path
 src_path = Path(__file__).parent / "src"
 sys.path.append(str(src_path))
 
-# Import and run the main app
-from src.app import main
-
 if __name__ == "__main__":
-    main() 
+    # Download model first
+    if download_model():
+        # Import and run the main app
+        from src.app import main
+        main()
+    else:
+        st.error("Cannot start the application without the model file.")