gxy commited on
Commit
d06d643
1 Parent(s): 717bb02

Update app.py

Browse files

FEAT: add hf_token

Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -8,9 +8,10 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
8
 
9
 
10
  _MODEL_PATH = 'IDEA-CCNL/Taiyi-BLIP-750M-Chinese'
11
-
12
- processor = BlipProcessor.from_pretrained(_MODEL_PATH)
13
- model = BlipForConditionalGeneration.from_pretrained(_MODEL_PATH).half().eval().to(device)
 
14
 
15
 
16
  def inference(raw_image, model_n, question, strategy):
 
8
 
9
 
10
  _MODEL_PATH = 'IDEA-CCNL/Taiyi-BLIP-750M-Chinese'
11
+ HF_TOKEN = os.getenv('HF_TOKEN')
12
+ processor = BlipProcessor.from_pretrained(_MODEL_PATH, use_auth_token=HF_TOKEN)
13
+ model = BlipForConditionalGeneration.from_pretrained(
14
+ _MODEL_PATH, use_auth_token=HF_TOKEN).half().eval().to(device)
15
 
16
 
17
  def inference(raw_image, model_n, question, strategy):