Update main.py

main.py

@@ -1,8 +1,8 @@
 import os
-cwd = os.getcwd()
-cachedir = cwd+'/cache'
-os.mkdir(cachedir)
-os.environ['TRANSFORMERS_CACHE'] = cachedir
+# cwd = os.getcwd()
+# cachedir = cwd+'/cache'
+# os.mkdir(cachedir)
+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',use_auth_token=False)
+model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
 
 class StringInput(BaseModel):
     sentence1: str