Spaces:

Chittrarasu
/

embedding_FastAPI

Runtime error

Chittrarasu commited on Feb 15

Commit

35b327b

1 Parent(s): 4e3ad22

deploy

Files changed (1) hide show

service/prediction_service.py CHANGED Viewed

@@ -19,4 +19,6 @@ if not os.path.exists(os.path.join(model_dir, 'config.json')):
     model.save(model_dir)
 else:
     print(f"Loading model from local directory: {model_dir}")
-    model = SentenceTransformer(model_dir)

     model.save(model_dir)
 else:
     print(f"Loading model from local directory: {model_dir}")
+    # model = SentenceTransformer(model_dir)
+    model = SentenceTransformer(model_dir, trust_remote_code=True)