Update main.py

main.py

@@ -2,7 +2,7 @@ import os
 # cwd = os.getcwd()
 # cachedir = cwd+'/cache'
 # os.mkdir(cachedir)
-os.environ['TRANSFORMERS_CACHE'] = "/app/yo/cache"
+# os.environ['TRANSFORMERS_CACHE'] = "/app/yo/cache"
 from fastapi import FastAPI, HTTPException
 import uvicorn
 from fastapi.middleware.cors import CORSMiddleware
@@ -13,7 +13,7 @@ app = FastAPI()
 
 
 # Load the SentenceTransformer model
-model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
+model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2',cache_folder="/app/yo")
 
 class StringInput(BaseModel):
     sentence1: str