joytheslothh commited on
Commit
a2659d0
·
1 Parent(s): 9f6646d

fix: replace tensorflow.keras with standalone keras for TF 2.16+

Browse files
bacsense_v2_package/inference.py CHANGED
@@ -20,7 +20,7 @@ warnings.filterwarnings("ignore")
20
  def _lazy_imports():
21
  from scipy.stats import skew
22
  from skimage.feature import local_binary_pattern, graycomatrix, graycoprops
23
- from tensorflow.keras.models import load_model
24
  return skew, local_binary_pattern, graycomatrix, graycoprops, load_model
25
 
26
 
 
20
  def _lazy_imports():
21
  from scipy.stats import skew
22
  from skimage.feature import local_binary_pattern, graycomatrix, graycoprops
23
+ from keras.models import load_model
24
  return skew, local_binary_pattern, graycomatrix, graycoprops, load_model
25
 
26
 
bacsense_v2_package/requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
  tensorflow>=2.12.0
 
2
  scikit-learn>=1.3.0
3
  scikit-image>=0.21.0
4
  opencv-python>=4.8.0
 
1
  tensorflow>=2.12.0
2
+ keras>=3.0.0
3
  scikit-learn>=1.3.0
4
  scikit-image>=0.21.0
5
  opencv-python>=4.8.0
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
- tensorflow-cpu>=2.15.0,<2.16.0
 
2
  scikit-learn>=1.3.0
3
  scikit-image>=0.21.0
4
  opencv-python-headless>=4.8.0
 
1
+ tensorflow-cpu>=2.15.0
2
+ keras>=3.0.0
3
  scikit-learn>=1.3.0
4
  scikit-image>=0.21.0
5
  opencv-python-headless>=4.8.0