Update main.py

main.py

@@ -24,10 +24,13 @@ logging.basicConfig(level=logging.INFO)
 ####################################
 # Text Generation Endpoint
 ####################################
-os.environ["TRANSFORMERS_CACHE"] = os.path.expanduser("~/huggingface_cache")
+import os
+os.environ["TRANSFORMERS_CACHE"] = "./cache"
+os.environ["HF_HOME"] = "./cache"
+
 TEXT_MODEL_NAME = "aubmindlab/aragpt2-base"
-text_tokenizer = AutoTokenizer.from_pretrained(TEXT_MODEL_NAME)
-text_model = AutoModelForCausalLM.from_pretrained(TEXT_MODEL_NAME)
+text_tokenizer = AutoTokenizer.from_pretrained(TEXT_MODEL_NAME, use_auth_token=True)
+text_model = AutoModelForCausalLM.from_pretrained(TEXT_MODEL_NAME,  use_auth_token=True)
 
 general_prompt_template = """
 أنت الآن نموذج لغة مخصص لتوليد نصوص عربية تعليمية بناءً على المادة والمستوى التعليمي. سيتم إعطاؤك مادة تعليمية ومستوى تعليمي، وعليك إنشاء نص مناسب بناءً على ذلك. النص يجب أن يكون: