Ryukijano commited on
Commit
bb85f4f
·
verified ·
1 Parent(s): b75198d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ import os
10
  hf_token = os.getenv("HF_TOKEN") # Fetch token from repository secrets
11
 
12
  # Set up Meta Llama 3.2 Vision model (using private model with token)
13
- llama_vision_model_id = "meta-llama/Llama-3.2-1B-Vision"
14
  vision_pipe = pipeline(
15
  "image-captioning", # Supports image captioning and image Q&A
16
  model=llama_vision_model_id,
 
10
  hf_token = os.getenv("HF_TOKEN") # Fetch token from repository secrets
11
 
12
  # Set up Meta Llama 3.2 Vision model (using private model with token)
13
+ llama_vision_model_id = "meta-llama/Llama-3.2-11B-Vision-Instruct"
14
  vision_pipe = pipeline(
15
  "image-captioning", # Supports image captioning and image Q&A
16
  model=llama_vision_model_id,