Update app.py

app.py

@@ -7,8 +7,9 @@ from werkzeug.utils import secure_filename
 # Suppress TensorFlow warnings
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
 
-# Use Keras 2.2.5 (compatible with Keras 2.0 models)
-from keras.models import load_model
+# Use TensorFlow 2.0 integrated Keras
+import tensorflow as tf
+from tensorflow.keras.models import load_model
 
 app = Flask(__name__)
 app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024  # 16MB max file size
@@ -17,15 +18,15 @@ 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 with Keras 2.2.5 (supports batch_shape)
-print("Loading model with Keras 2.2.5...")
+# Load the model with TensorFlow 2.0
+print("Loading model with TensorFlow 2.0...")
 import warnings
 warnings.filterwarnings('ignore')
 
 try:
-    # Keras 2.2.5 natively supports batch_shape parameter
+    # TensorFlow 2.0 should handle Functional models
     model = load_model('cancer_model.h5', compile=False)
-    print("✓ Model loaded successfully with Keras 2.2.5!")
+    print("✓ Model loaded successfully with TensorFlow 2.0!")
 except Exception as e:
     print(f"❌ Error loading model: {e}")
     import traceback