Add latest updates and clean up

fix_model_config.py

@@ -1,30 +0,0 @@
-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.")

requirements.txt

@@ -1,6 +1,10 @@
+tensorflow==2.18.0
 streamlit==1.32.0
-tensorflow==2.15.0
-numpy==1.24.3
-opencv-python==4.9.0.80
-matplotlib==3.7.1
-Pillow==9.5.0
+ml-dtypes==0.4.0
+tensorboard==2.18.0
+numpy
+pillow
+opencv-python
+scikit-learn
+matplotlib
+seaborn

src/models/trashsort_cnn.h5

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