Bils commited on
Commit
60ed470
·
verified ·
1 Parent(s): f1c8934

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ device_id = 0 if torch.cuda.is_available() else -1
16
 
17
  captioning_pipeline = pipeline(
18
  "image-to-text",
19
- model="nlpconnect/vit-gpt2-image-captioning",
20
  device=device_id
21
  )
22
 
 
16
 
17
  captioning_pipeline = pipeline(
18
  "image-to-text",
19
+ model="Salesforce/blip-image-captioning-large", #
20
  device=device_id
21
  )
22