Update app.py

app.py

@@ -12,7 +12,7 @@ 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-mid-training" # "BounharAbdelaziz/Al-Atlas-LLM-0.5B" #"atlasia/Al-Atlas-LLM"
+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)