hgdgng commited on
Commit
d35db82
1 Parent(s): d0291e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ ckpt = "meta-llama/Llama-3.2-11B-Vision-Instruct"
11
  model = MllamaForConditionalGeneration.from_pretrained(ckpt,
12
  torch_dtype=torch.bfloat16).to("cuda")
13
  processor = AutoProcessor.from_pretrained(ckpt)
14
- tokenizer = AutoTokenizer.from_pretrained( "meta-llama/Llama-3.2-11B-Vision-Instruct", token=HF_TOKEN )
15
 
16
  @spaces.GPU
17
  def bot_streaming(message, history, max_new_tokens=250):
 
11
  model = MllamaForConditionalGeneration.from_pretrained(ckpt,
12
  torch_dtype=torch.bfloat16).to("cuda")
13
  processor = AutoProcessor.from_pretrained(ckpt)
14
+ tokenizer = AutoTokenizer.from_pretrained( "meta-llama/Llama-3.2-11B-Vision-Instruct", token=HGHugging )
15
 
16
  @spaces.GPU
17
  def bot_streaming(message, history, max_new_tokens=250):