Update main.py

main.py

@@ -2,12 +2,11 @@ from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel, root_validator
 from transformers import AutoModel
 from typing import List
-import os, platform
-os.environ['TRANSFORMERS_CACHE'] = '/blabla/cache/' 
+import os, platform 
 
 
 
-model = AutoModel.from_pretrained('moraxgiga/jina', trust_remote_code=True)
+model = AutoModel.from_pretrained('moraxgiga/jina',cache_dir="new_cache_dir/",  trust_remote_code=True)
 
 app = FastAPI()