Update src/api/index.py

src/api/index.py

@@ -32,6 +32,8 @@ from pydantic import BaseModel
 from src.classes import hugging_face_utils as hf
 from src.classes import semantic_search_engine as ss
 from src.utils import default_variables as dv
+import os
+from pathlib import Path
 
 logger = logging.getLogger(__name__)
 logging.basicConfig(
@@ -62,6 +64,12 @@ app = FastAPI(
 
 # -------------------------------- CLASSES ------------------------------------
 
+# Creating directory
+cache_dir = Path(".").resolve().joinpath("cache")
+cache_dir.mkdir(exist_ok=True,parents=True,)
+
+os.environ['SENTENCE_TRANSFORMERS_HOME'] = str(cache_dir)
+
 
 class QueryParams(BaseModel):
     input_query: str