Fix Keras model loading compatibility (compile=False)

app.py

@@ -13,8 +13,10 @@ from PIL import Image
 
 # Import TensorFlow
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
+os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
+
 import tensorflow as tf
-from tensorflow.keras.models import load_model
+from tensorflow import keras
 from tensorflow.keras.applications.inception_v3 import preprocess_input
 
 app = Flask(__name__)
@@ -32,10 +34,20 @@ def load_models():
     model_dir = "models"
     
     try:
-        # Load Keras model
+        # Load Keras model with compile=False to avoid custom layer issues
         model_path = os.path.join(model_dir, "batik_model.keras")
-        model = load_model(model_path)
+        model = keras.models.load_model(model_path, compile=False)
+        
+        # Compile manually with simple config
+        model.compile(
+            optimizer='adam',
+            loss='categorical_crossentropy',
+            metrics=['accuracy']
+        )
+        
         print(f" Loaded Keras model from {model_path}")
+        print(f"   Model input shape: {model.input_shape}")
+        print(f"   Model output shape: {model.output_shape}")
         
         # Load class names
         classes_path = os.path.join(model_dir, "batik_classes.json")
@@ -74,9 +86,11 @@ def index():
     """API info endpoint"""
     return jsonify({
         "name": "Batik Classifier API",
-        "model": "Fine-tuned InceptionV3",
+        "model": "Fine-tuned InceptionV3 (Keras)",
         "classes": len(class_names) if class_names else 0,
         "accuracy": config.get('val_accuracy', 0) if config else 0,
+        "train_accuracy": config.get('train_accuracy', 0) if config else 0,
+        "epochs": config.get('epochs', 0) if config else 0,
         "endpoints": {
             "/": "API info",
             "/predict": "POST - Classify batik image",
@@ -161,7 +175,10 @@ def predict():
         }), 500
 
 # Load models on startup
-print("Loading models...")
+print("=" * 60)
+print("Loading Batik Classifier Models...")
+print("=" * 60)
+
 if load_models():
     print(" All models loaded successfully!")
 else:

requirements.txt

@@ -1,6 +1,6 @@
 Flask==3.0.3
 flask-cors==5.0.0
-tensorflow==2.16.1
+tensorflow==2.18.0
 scikit-learn==1.4.2
 numpy>=1.23.5,<2.0.0
 pillow==10.2.0