Update main.py

main.py

@@ -1,3 +1,6 @@
+os.environ["HF_HOME"] = "/tmp/huggingface"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/transformers_cache"
+
 from fastapi import FastAPI, File, UploadFile
 from fastapi.responses import StreamingResponse
 from fastapi.middleware.cors import CORSMiddleware
@@ -176,8 +179,7 @@ async def retrieve_top_k_hybrid(query, k, sem_weight,syn_weight,bm25):
 
     return top_results["Chunks"].to_list()
 
-os.environ["TRANSFORMERS_CACHE"] = "/tmp/transformers_cache"
-os.environ["HF_HOME"] = "/tmp/huggingface"
+
 os.makedirs("/tmp/transformers_cache", exist_ok=True)
 
 model = SentenceTransformer("abhinand/MedEmbed-large-v0.1")