Upload app.py

app.py

@@ -1,8 +1,14 @@
 import os
 
-import os
-os.environ['TRANSFORMERS_CACHE'] = '/home/user/.cache/huggingface'
-os.environ['HF_HOME'] = '/home/user/.cache/huggingface'
+cache_dir = os.path.join(os.getcwd(), ".cache")
+try:
+    os.makedirs(cache_dir, exist_ok=True)
+except PermissionError:
+    cache_dir = "/tmp/.cache"
+    os.makedirs(cache_dir, exist_ok=True)
+
+os.environ['TRANSFORMERS_CACHE'] = cache_dir
+os.environ['HF_HOME'] = cache_dir
 
 
 from fastapi import FastAPI, Header, HTTPException, Body