Update main.py

main.py

@@ -24,7 +24,7 @@ logging.basicConfig(level=logging.INFO)
 ####################################
 # Text Generation Endpoint
 ####################################
-os.environ["TRANSFORMERS_CACHE"] = "/app/cache"
+os.environ["TRANSFORMERS_CACHE"] = os.path.expanduser("~/huggingface_cache")
 TEXT_MODEL_NAME = "aubmindlab/aragpt2-base"
 text_tokenizer = AutoTokenizer.from_pretrained(TEXT_MODEL_NAME)
 text_model = AutoModelForCausalLM.from_pretrained(TEXT_MODEL_NAME)