Update app.py

app.py

@@ -6,8 +6,23 @@ import numpy as np
 import tensorflow as tf
 from PIL import Image
 
-model_path = "waste_classifier_final_fixed.h5"
-model = tf.keras.models.load_model(model_path, compile=False)
+# ✅ Custom fix for batch_shape compatibility
+from tensorflow.keras.layers import InputLayer
+
+class CompatibleInputLayer(InputLayer):
+    def __init__(self, *args, **kwargs):
+        kwargs.pop('batch_shape', None)
+        kwargs.pop('optional', None)
+        super().__init__(*args, **kwargs)
+
+# Load model with custom object
+model_path = "waste_classifier_patched.h5"
+model = tf.keras.models.load_model(
+    model_path,
+    compile=False,
+    custom_objects={"InputLayer": CompatibleInputLayer}
+)
+print("✅ Model loaded!")
 
 labels = ["Cardboard", "Glass", "Metal", "Paper", "Plastic", "Trash"]