upload

embedding_model.py

@@ -14,7 +14,7 @@ class HuggingfaceSentenceTransformerModel(Embedding_Model):
     def __init__(self, model_name=EMBEDDING_MODEL) -> None:
         super().__init__(model_name)
         
-        self.model = SentenceTransformer(model_name)
+        self.model = SentenceTransformer(model_name, cache_folder="/app/ckpt/")
 
     def __call__(self, text) -> None:
         return self.model.encode(text)