| import gradio | |
| from transformers import pipeline | |
| _pipeline = pipeline("image-to-text", "Salesforce/blip-image-captioning-base") | |
| def launch(pipeline_input): | |
| out = _pipeline(pipeline_input) | |
| return out[0]["generated_text"] | |
| _interface = gradio.Interface(launch, inputs=gradio.Image(type="pil"), outputs="text") | |
| _interface.launch() | |