Update app.py

app.py

@@ -50,7 +50,8 @@ def download_model_from_drive():
     
     if not os.path.exists(model_path):
         # Extract the file ID from the sharing URL
-        url = "https://drive.google.com/file/d/1XObpqG8qZ7YUyiRKbpVvxX11yQSK8Y_3/view?usp=sharing"
+        file_id = "1XObpqG8qZ7YUyiRKbpVvxX11yQSK8Y_3"
+        url = f"https://drive.google.com/uc?id={file_id}"
         try:
             gdown.download(url, model_path, quiet=False)
             st.success("Model downloaded successfully from Google Drive.")
@@ -91,8 +92,8 @@ def load_model():
                 st.success("Model weights loaded successfully!")
                 return base_model
             except Exception as e:
-                # st.error(f"Error loading weights: {e}")
-                # st.info("Using base pretrained model instead")
+                st.error(f"Error loading weights: {e}")
+                st.info("Using base pretrained model instead")
                 return base_model
         else:
             st.warning("Using base pretrained model since download failed")