rohansampath commited on
Commit
dbd35bb
·
verified ·
1 Parent(s): b04caf6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ import base64
12
  # ---------------------------------------------------------------------------
13
  # 1. Define model name and load model/tokenizer
14
  # ---------------------------------------------------------------------------
15
- model_name = "meta-llama/Llama-3.2-1B-Instruct" # fictional placeholder
16
 
17
  tokenizer = AutoTokenizer.from_pretrained(model_name)
18
  model = AutoModelForCausalLM.from_pretrained(model_name)
 
12
  # ---------------------------------------------------------------------------
13
  # 1. Define model name and load model/tokenizer
14
  # ---------------------------------------------------------------------------
15
+ model_name = "mistralai/Mistral-7B-Instruct-v0.3" # fictional placeholder
16
 
17
  tokenizer = AutoTokenizer.from_pretrained(model_name)
18
  model = AutoModelForCausalLM.from_pretrained(model_name)