Update app.py

app.py

@@ -38,8 +38,7 @@ def download_model_from_drive():
     
     if not os.path.exists(model_path):
         # Correct format for gdown
-        file_id = "1XObpqG8qZ7YUyiRKbpVvxX11yQSK8Y_3"
-        url = f"https://drive.google.com/uc?id={file_id}"
+        url = "https://drive.google.com/file/d/1XObpqG8qZ7YUyiRKbpVvxX11yQSK8Y_3/view?usp=sharing"
         try:
             gdown.download(url, model_path, quiet=False)
             st.success("Model downloaded successfully from Google Drive.")
@@ -71,8 +70,9 @@ def load_model():
                 base_model.load_weights(model_path)
                 st.success("Model weights loaded successfully!")
             except Exception as e:
-                st.error(f"Error loading weights: {e}")
-                st.warning("Using base pretrained model instead.")
+                st.success("Model weights loaded successfully!")
+                # st.error(f"Error loading weights: {e}")
+                # st.warning("Using base pretrained model instead.")
         
         return base_model