Vijayrathank commited on
Commit
59ec3ea
·
verified ·
1 Parent(s): 7ffbf34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import os
8
  hf_token = os.getenv("HF_TOKEN")
9
  login(token=hf_token)
10
 
11
- model_id = "mistralai/Mistral-7B-Instruct-v0.3" # small enough to run locally on CPU
12
  tokenizer = AutoTokenizer.from_pretrained(model_id, token=hf_token)
13
  model = AutoModelForCausalLM.from_pretrained(model_id, token=hf_token)
14
 
 
8
  hf_token = os.getenv("HF_TOKEN")
9
  login(token=hf_token)
10
 
11
+ model_id = "meta-llama/Llama-3.2-1B" # small enough to run locally on CPU
12
  tokenizer = AutoTokenizer.from_pretrained(model_id, token=hf_token)
13
  model = AutoModelForCausalLM.from_pretrained(model_id, token=hf_token)
14