Update app.py

app.py

@@ -18,17 +18,28 @@ print("Loading model...")
 import warnings
 warnings.filterwarnings('ignore')
 
-# Custom loader for compatibility with different TensorFlow versions
+# Custom object scope to handle old Keras model format
+from tensorflow.keras.layers import InputLayer
+import tensorflow.keras.backend as K
+
+# Patch for old model format compatibility
+def custom_input_layer(*args, **kwargs):
+    # Convert old 'batch_shape' to new 'shape'
+    if 'batch_shape' in kwargs:
+        batch_shape = kwargs.pop('batch_shape')
+        if batch_shape is not None and len(batch_shape) > 1:
+            kwargs['shape'] = batch_shape[1:]
+    return InputLayer(*args, **kwargs)
+
+# Try loading with custom objects
 try:
-    model = load_model('cancer_model.h5', compile=False)
+    with tf.keras.utils.custom_object_scope({'InputLayer': custom_input_layer}):
+        model = load_model('cancer_model.h5', compile=False)
+    print("Model loaded successfully!")
 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!")
+    print(f"Error loading model: {e}")
+    print("Please ensure the model file is compatible with TensorFlow 2.13.0")
+    raise
 
 def resize_with_padding(img, target_size):
     """Resize image while maintaining aspect ratio and add padding"""