File size: 319 Bytes
b048ad3
 
 
 
 
e253f7b
 
b048ad3
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
import gradio as gr
from transformers import pipeline

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

def launch(inputs):
    output = pipe(inputs)
    return output[0]['generated_text']

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

iface.launch()