Spaces:

fau
/

videoxity

Running

zamal commited on 25 days ago

Commit

05fa324

1 Parent(s): 913aabe

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -26,7 +26,8 @@ if not HF_TOKEN:
 captioner = pipeline(
     "image-to-text",
     model="Salesforce/blip-image-captioning-base",
-    device="cpu"
 )
 vl_pipeline = pipeline(

 captioner = pipeline(
     "image-to-text",
     model="Salesforce/blip-image-captioning-base",
+    device="cpu",
+    use_safetensors=True
 )
 vl_pipeline = pipeline(