Update main.py

main.py

@@ -24,9 +24,9 @@ logging.basicConfig(level=logging.INFO)
 ####################################
 # Text Generation Endpoint
 ####################################
-import os
-os.environ["TRANSFORMERS_CACHE"] = "./cache"
-os.environ["HF_HOME"] = "./cache"
+import tempfile
+os.environ["TRANSFORMERS_CACHE"] = tempfile.mkdtemp()
+
 HF_TOKEN = os.getenv("manhal")
 
 TEXT_MODEL_NAME = "aubmindlab/aragpt2-base"