Update app.py

app.py

@@ -12,10 +12,10 @@ print(f'[INFO] Using device: {device}')
 token = os.environ['TOKEN']
 
 # Load the pretrained model and tokenizer
-MODEL_NAME = "BounharAbdelaziz/Al-Atlas-LLM-0.5B" #"atlasia/Al-Atlas-LLM"
+MODEL_NAME = "atlasia/Al-Atlas-LLM" # "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)
+model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, token=token).to(device)
 
 # Predefined examples
 examples = [