Update app.py

app.py

@@ -24,21 +24,34 @@ os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'
 # Try loading with h5py directly for better compatibility
 import h5py
 
-# Load model with old TensorFlow/Keras compatibility
+# Custom InputLayer to handle old 'batch_shape' parameter
+from tensorflow.python.keras.engine.input_layer import InputLayer as OriginalInputLayer
+
+class CustomInputLayer(OriginalInputLayer):
+    def __init__(self, **kwargs):
+        # Convert old 'batch_shape' to new 'input_shape'
+        if 'batch_shape' in kwargs:
+            batch_shape = kwargs.pop('batch_shape')
+            if batch_shape and len(batch_shape) > 1:
+                kwargs['input_shape'] = batch_shape[1:]
+        # Remove any other problematic params
+        kwargs.pop('batch_input_shape', None)
+        super(CustomInputLayer, self).__init__(**kwargs)
+
+# Register custom layer
+custom_objects = {'InputLayer': CustomInputLayer}
+
+# Load model with custom objects
 try:
-    # Direct loading with old Keras
-    from keras.models import load_model as keras_load_model
-    model = keras_load_model('cancer_model.h5', compile=False)
-    print("✓ Model loaded successfully with Keras 2.4.3!")
+    model = tf.keras.models.load_model('cancer_model.h5', 
+                                      custom_objects=custom_objects, 
+                                      compile=False)
+    print("✓ Model loaded successfully!")
 except Exception as e:
-    print(f"Keras loader failed: {e}")
-    try:
-        # Try TensorFlow loader
-        model = tf.keras.models.load_model('cancer_model.h5', compile=False)
-        print("✓ Model loaded successfully with TensorFlow!")
-    except Exception as e2:
-        print(f"Error: {e2}")
-        raise
+    print(f"Error loading model: {e}")
+    print("\n⚠️ Model file is incompatible with TensorFlow 2.4.1")
+    print("Your model appears to be from TensorFlow 1.x or very early 2.x")
+    raise
 
 def resize_with_padding(img, target_size):
     """Resize image while maintaining aspect ratio and add padding"""