Update app.py

app.py

@@ -13,9 +13,21 @@ app.config['UPLOAD_FOLDER'] = 'uploads'
 # Create uploads folder if it doesn't exist
 os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
 
-# Load the model
+# Load the model with custom objects to handle compatibility
 print("Loading model...")
-model = load_model('cancer_model.h5', compile=False)
+import warnings
+warnings.filterwarnings('ignore')
+
+# Custom loader for compatibility with different TensorFlow versions
+try:
+    model = load_model('cancer_model.h5', compile=False)
+except Exception as e:
+    print(f"First attempt failed: {e}")
+    print("Trying with safe mode...")
+    # Try loading with safe mode
+    from tensorflow import keras
+    model = keras.models.load_model('cancer_model.h5', compile=False, safe_mode=False)
+
 print("Model loaded successfully!")
 
 def resize_with_padding(img, target_size):