Update app.py

app.py

@@ -14,8 +14,8 @@ token = os.environ['TOKEN']
 # Load the pretrained model and tokenizer
 MODEL_NAME = "atlasia/Al-Atlas-0.5B" # "atlasia/Al-Atlas-LLM-mid-training" # "BounharAbdelaziz/Al-Atlas-LLM-0.5B" #"atlasia/Al-Atlas-LLM"
 
-tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME) # , token=token
-model = AutoModelForCausalLM.from_pretrained(MODEL_NAME).to(device)
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME,token=token) # , token=token
+model = AutoModelForCausalLM.from_pretrained(MODEL_NAME,token=token).to(device)
 
 # Fix tokenizer padding
 if tokenizer.pad_token is None: