Update main.py

main.py

@@ -1,4 +1,8 @@
 import os
+cwd = os.getcwd()
+cachedir = cwd+'/cache'
+os.mkdir(cachedir)
+os.environ['TRANSFORMERS_CACHE'] = cachedir
 from fastapi import FastAPI, HTTPException
 import uvicorn
 from fastapi.middleware.cors import CORSMiddleware
@@ -6,7 +10,7 @@ from pydantic import BaseModel
 from sentence_transformers import SentenceTransformer, util
 
 app = FastAPI()
-current_directory = os.getcwd()
+
 
 # Load the SentenceTransformer model
 model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2',use_auth_token=False)