BounharAbdelaziz commited on
Commit
f82e570
·
verified ·
1 Parent(s): cf3092a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,8 +14,8 @@ token = os.environ['TOKEN']
14
  # Load the pretrained model and tokenizer
15
  MODEL_NAME = "BounharAbdelaziz/Al-Atlas-LLM-0.5B" #"atlasia/Al-Atlas-LLM"
16
 
17
- tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, token=token)
18
- model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, token=token).to(device)
19
 
20
  # Predefined examples
21
  examples = [
 
14
  # Load the pretrained model and tokenizer
15
  MODEL_NAME = "BounharAbdelaziz/Al-Atlas-LLM-0.5B" #"atlasia/Al-Atlas-LLM"
16
 
17
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)#, token=token)
18
+ model = AutoModelForCausalLM.from_pretrained(MODEL_NAME).to(device)
19
 
20
  # Predefined examples
21
  examples = [