File size: 341 Bytes
21dc45f
 
 
 
93ab40f
21dc45f
 
 
 
93ab40f
21dc45f
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import gradio as gr
from transformers import pipeline

pipe = pipeline(
    "image-text-to-text",
    model="Salesforce/blip-image-captioning-base"
)

def launch(input):
    out = pipe(input, text="")
    return out[0]['generated_text']

iface = gr.Interface(
    launch,
    inputs=gr.Image(type='pil'),
    outputs="text"
)

iface.launch()