Update app.py

app.py

@@ -6,18 +6,21 @@ from PIL import Image
 import zipfile
 import gdown
 
-# Path to your zipped model file
-ZIP_MODEL_URL = 'https://drive.google.com/uc?id=1-4p6AZBkooWL1rhN9WwIrfd9fJbhzY0e'  # Google Drive link to the model
-UNZIPPED_MODEL_PATH = '/app/your_trained_model.keras'  # Path where the model will be extracted
+# Path to your zipped model file (this will be the local path after downloading)
+ZIP_MODEL_PATH = '/app/your_trained_model_resnet50.keras.zip'  
+UNZIPPED_MODEL_PATH = '/app/your_trained_model_resnet50.keras'  # Path where the model will be extracted
 
-# Download the model from Google Drive if it hasn't been downloaded already
-if not os.path.exists(UNZIPPED_MODEL_PATH):
-    # Download model from Google Drive
-    output = '/app/your_trained_model_resnet50.keras.zip'  # Path where model will be saved
-    gdown.download(ZIP_MODEL_URL, output, quiet=False)
+# Google Drive link to the model file
+MODEL_URL = 'https://drive.google.com/uc?export=download&id=1-4p6AZBkooWL1rhN9WwIrfd9fJbhzY0e'
+
+# Download the model if it doesn't exist
+if not os.path.exists(ZIP_MODEL_PATH):
+    gdown.download(MODEL_URL, ZIP_MODEL_PATH, quiet=False)
+    print(f"Model downloaded to {ZIP_MODEL_PATH}")
 
-    # Unzip the model
-    with zipfile.ZipFile(output, 'r') as zip_ref:
+# Unzip the model if it hasn't been unzipped already
+if not os.path.exists(UNZIPPED_MODEL_PATH):
+    with zipfile.ZipFile(ZIP_MODEL_PATH, 'r') as zip_ref:
         zip_ref.extractall('/app')
         print(f"Model unzipped to {UNZIPPED_MODEL_PATH}")