Update app.py

app.py

@@ -11,6 +11,8 @@ import numpy as np
 import tensorflow as tf
 from huggingface_hub import snapshot_download
 
+# check the tensoflow version
+print(f"tensorflow version: {tf.__version__}")
 
 # global variables
 # predictions from:
@@ -72,8 +74,9 @@ def download_keras_model():
         print(f"Download dir: {download_dir}")
         keras_binary_path = os.path.join(download_dir, keras_binary_label)
         print(f"Keras binary label: {keras_binary_path}")
-        keras_binary_model = tf.keras.models.load(keras_binary_path)
-
+        # this load() does not work in HF
+        #keras_binary_model = tf.keras.models.load(keras_binary_path)
+        keras_binary_model = tf.keras.saving.load_model(keras_binary_path)
     # local testing
     # check if the model exists
     # binary label
@@ -86,7 +89,7 @@ def download_keras_model():
         )
         if not os.path.exists(model_path):
             print(f"Model not found: {model_path}")
-        # load keras model
+        # load local keras model
         keras_binary_model = tf.keras.models.load_model(model_path)
 
     # Check with model loaded
@@ -102,8 +105,7 @@ def download_keras_model():
         )
         if not os.path.exists(model_path):
             print(f"Model not found: {model_path}")
-        # load keras model
-
+        # load local keras model
         keras_multi_model = tf.keras.models.load_model(model_path)
 
     # In HF space, load model from repository
@@ -114,7 +116,9 @@ def download_keras_model():
         print(f"Download dir: {download_dir}")
         keras_multi_path = os.path.join(download_dir, keras_multilabel)
         print(f"Keras multi label: {keras_multi_path}")
-        keras_multi_model = tf.keras.models.load(keras_multi_path)
+        # load() does not work in HF
+        #keras_multi_model = tf.keras.models.load(keras_multi_path)
+        keras_multi_model = tf.keras.saving.load_model(keras_multi_path)
 
     # Check with model loaded
     print(f"\nLoaded model: {keras_multi_model.name}")