IanRonk commited on
Commit
c0d4bf9
·
1 Parent(s): 7de2a15

Try other model

Browse files
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("RNN_model.keras")
10
 
11
 
12
  def query_embeddings(texts):
 
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):
functions/tf_format.keras ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5af590bbc6d50b5ca00d2f7cdca06d2e6c8ef94dd6e09019c69b75e816ca5d05
3
+ size 3977504
tf_format.keras ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5af590bbc6d50b5ca00d2f7cdca06d2e6c8ef94dd6e09019c69b75e816ca5d05
3
+ size 3977504