Subhajit Chakraborty commited on
Commit
a49b078
·
1 Parent(s): 24eab42

update app.py(3)

Browse files
Files changed (1) hide show
  1. src/app.py +1 -1
src/app.py CHANGED
@@ -25,7 +25,7 @@ def load_model():
25
  os.environ["SENTENCE_TRANSFORMERS_HOME"] = "/tmp/huggingface/sentence_transformers"
26
 
27
  os.makedirs("/tmp/huggingface", exist_ok=True)
28
- return SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2", cache_dir="/tmp/huggingface")
29
 
30
 
31
  embedder = load_model()
 
25
  os.environ["SENTENCE_TRANSFORMERS_HOME"] = "/tmp/huggingface/sentence_transformers"
26
 
27
  os.makedirs("/tmp/huggingface", exist_ok=True)
28
+ return SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2", cache_folder="/tmp/huggingface")
29
 
30
 
31
  embedder = load_model()