Spaces:

robinroy03
/

fury-embeddings-endpoint

Sleeping

robinroy03 commited on Jun 5

Commit

26d86c2

•

1 Parent(s): 05411e6

main.py - download model

Files changed (1) hide show

main.py CHANGED Viewed

@@ -5,8 +5,8 @@ from sentence_transformers import SentenceTransformer
 import os
 app = Flask(__name__)
-model = SentenceTransformer('embedding_model', trust_remote_code=True)
 os.environ["TRANSFORMERS_CACHE"] = "/.cache"
 @app.route("/embedding", methods=['POST'])
 def embedding():

 import os
 app = Flask(__name__)
 os.environ["TRANSFORMERS_CACHE"] = "/.cache"
+model = SentenceTransformer('Alibaba-NLP/gte-large-en-v1.5', trust_remote_code=True)
 @app.route("/embedding", methods=['POST'])
 def embedding():