import gradio as gr from transformers import pipeline pipe = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base") def launch(input): output = pipe(input) return output[0]["generated_text"] iface = gr.Interface(launch, inputs=gr.Image(type='pil'), outputs="text") iface.launch()