Update app.py

app.py

@@ -42,17 +42,19 @@ class FixedInputLayer(InputLayer):
             del kwargs['dtype']
         super().__init__(**kwargs)
 
-# 3. Xử lý Augmentation Layers (Xử lý lỗi: data_format & dtype policy)
+# 3. Xử lý Augmentation Layers (Xử lý lỗi: data_format, dtype & value_range)
 def fix_augmentation_layer(LayerClass):
     class FixedLayer(LayerClass):
         def __init__(self, **kwargs):
-            # Xóa tham số data_format gây lỗi
-            if 'data_format' in kwargs:
-                del kwargs['data_format']
+            # Danh sách các tham số gây lỗi tương thích giữa Keras 3 và 2
+            # 'value_range': Gây lỗi ở RandomContrast
+            # 'data_format': Gây lỗi ở RandomFlip
+            # 'dtype': Gây lỗi DTypePolicy
+            ignore_keys = ['data_format', 'dtype', 'value_range']
             
-            # Xóa tham số dtype gây lỗi (DTypePolicy)
-            if 'dtype' in kwargs:
-                del kwargs['dtype']
+            for key in ignore_keys:
+                if key in kwargs:
+                    del kwargs[key]
                 
             super().__init__(**kwargs)
     return FixedLayer