Update app.py

app.py

@@ -4,14 +4,20 @@ import numpy as np
 import tensorflow as tf
 from PIL import Image
 import zipfile
+import gdown
 
 # Path to your zipped model file
-ZIP_MODEL_PATH = '/app/your_trained_model_resnet50.keras.zip'  # Updated path for the new model
-UNZIPPED_MODEL_PATH = '/app/your_trained_model_resnet50.keras'  # Path where the model will be extracted
+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
 
-# Unzip the model if it hasn't been unzipped already
+# Download the model from Google Drive if it hasn't been downloaded already
 if not os.path.exists(UNZIPPED_MODEL_PATH):
-    with zipfile.ZipFile(ZIP_MODEL_PATH, 'r') as zip_ref:
+    # 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)
+
+    # Unzip the model
+    with zipfile.ZipFile(output, 'r') as zip_ref:
         zip_ref.extractall('/app')
         print(f"Model unzipped to {UNZIPPED_MODEL_PATH}")