IanRonk commited on
Commit
9654080
·
1 Parent(s): e4b8c2d
Files changed (1) hide show
  1. functions/model_infer.py +1 -1
functions/model_infer.py CHANGED
@@ -6,7 +6,7 @@ from tensorflow.keras.models import load_model
6
 
7
  headers = {"Authorization": f"Bearer {os.environ['HF_Token']}"}
8
 
9
- model = load_model("tf_format.keras")
10
 
11
 
12
  def query_embeddings(texts):
 
6
 
7
  headers = {"Authorization": f"Bearer {os.environ['HF_Token']}"}
8
 
9
+ model = load_model("./model.keras")
10
 
11
 
12
  def query_embeddings(texts):