Spaces:

dolphinium
/

blip-image-captioning-api

Sleeping

dolphinium commited on Aug 24, 2024

Commit

4545740

verified ·

1 Parent(s): c643bd8

set clean_up_tokenization_spaces to False

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
-pipe = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base", clean_up_tokenization_spaces=True)
 def launch(input):
   out = pipe(input)

 import gradio as gr
 from transformers import pipeline
+pipe = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base", clean_up_tokenization_spaces=False)
 def launch(input):
   out = pipe(input)