Model new

functions/model_infer.py

@@ -1,12 +1,13 @@
-from tensorflow.keras.preprocessing.sequence import pad_sequences
-import tensorflow as tf
+from keras.preprocessing.sequence import pad_sequences
+
+# import tensorflow as tf
 import os
 import requests
-from tensorflow.keras.models import load_model
+from keras.models import load_model
 
 headers = {"Authorization": f"Bearer {os.environ['HF_Token']}"}
 
-model = load_model("./model.keras")
+model = load_model("./RNN_model.keras")
 
 
 def query_embeddings(texts):