Spaces:

ybelkada
/

dlai-test-2

Sleeping

ybelkada commited on Feb 20, 2024

Commit

cca1333

verified ·

1 Parent(s): 16eb2b2

Update app.py

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")
 def launch(input):
     out = pipe(input)

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