Update trained model trashsort_cnn.h5

fix_model_config.py

@@ -0,0 +1,30 @@
+import h5py
+import json
+
+model_path = 'src/models/trashsort_cnn.h5'
+
+with h5py.File(model_path, 'r+') as f:
+    config_raw = f.attrs.get('model_config')
+    
+    # If bytes, decode; if string, use as is
+    if isinstance(config_raw, bytes):
+        config_str = config_raw.decode('utf-8')
+    else:
+        config_str = config_raw
+
+    config = json.loads(config_str)
+
+    def fix_config(cfg):
+        if isinstance(cfg, dict):
+            if cfg.get('class_name') == 'InputLayer' and 'config' in cfg and 'batch_shape' in cfg['config']:
+                cfg['config'].pop('batch_shape')
+            for v in cfg.values():
+                fix_config(v)
+        elif isinstance(cfg, list):
+            for item in cfg:
+                fix_config(item)
+
+    fix_config(config)
+    f.attrs['model_config'] = json.dumps(config).encode('utf-8')
+
+print("Model config fixed successfully.")

src/models/trashsort_cnn.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f1f06aa9fca094e0bf771ea2de19367e9a679d50b42cb37a8bde2dacf72948ce
-size 9646624
+oid sha256:4629b65fb48a0eaa1c86ef9e12886d9b53c6d709499ca3ab7580a8475b78371c
+size 9783944