add cache dir

main.py

@@ -13,12 +13,20 @@ from PIL import Image
 from cachetools import Cache
 import torch
 import torch.nn.functional as F
+import os
 from models import (
     FileImageDetectionResponse,
     UrlImageDetectionResponse,
     ImageUrlsRequest,
 )
 
+PATH = '/home/ahmadzen/.cache/huggingface/'
+DATASETPATH = '/home/ahmadzen/.cache/huggingface/datasests/'
+MODELPATH = '/home/ahmadzen/.cache/huggingface/model'
+os.environ['TRANSFORMERS_CACHE'] = MODELPATH
+os.environ['HF_HOME'] = PATH
+os.environ['HF_DATASETS_CACHE'] = DATASETPATH
+os.environ['TORCH_HOME'] = PATH
 
 app = FastAPI()