Update app.py

app.py

@@ -15,7 +15,24 @@ os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
 
 from tensorflow.keras.models import load_model
 from tensorflow.keras.preprocessing.sequence import pad_sequences
-from tensorflow.keras.layers import Embedding
+import tensorflow as tf
+
+def make_safe_layer(layer_class):
+    class SafeLayer(layer_class):
+        def __init__(self, **kwargs):
+            # Hapus racunnya dari memori
+            kwargs.pop('quantization_config', None)
+            super().__init__(**kwargs)
+    return SafeLayer
+
+safe_objects = {
+    'Embedding': make_safe_layer(tf.keras.layers.Embedding),
+    'Dense': make_safe_layer(tf.keras.layers.Dense),
+    'LSTM': make_safe_layer(tf.keras.layers.LSTM),
+    'Dropout': make_safe_layer(tf.keras.layers.Dropout),
+    'SpatialDropout1D': make_safe_layer(tf.keras.layers.SpatialDropout1D),
+    'Bidirectional': make_safe_layer(tf.keras.layers.Bidirectional)
+}
 
 class SafeEmbedding(Embedding):
     def __init__(self, **kwargs):
@@ -47,7 +64,7 @@ def load_pkl(path):
 def load_keras(path):
     try: 
         print(f"Mencoba load model Keras: {path}")
-        return load_model(path, custom_objects={'Embedding': SafeEmbedding})
+        return load_model(path, custom_objects=safe_objects)
     except Exception as e: 
         print(f"❌ ERROR KERAS ({path}): {str(e)}")
         return None