zamal commited on
Commit
b7128c2
·
1 Parent(s): e98572f

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -2
main.py CHANGED
@@ -26,8 +26,7 @@ if not HF_TOKEN:
26
  captioner = pipeline(
27
  "image-to-text",
28
  model="Salesforce/blip-image-captioning-base",
29
- device="cpu",
30
- use_safetensors=True
31
  )
32
 
33
  vl_pipeline = pipeline(
 
26
  captioner = pipeline(
27
  "image-to-text",
28
  model="Salesforce/blip-image-captioning-base",
29
+ device="cpu"
 
30
  )
31
 
32
  vl_pipeline = pipeline(