Latest checkpoint

app.py

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