Spaces:
Running
Running
import gradio as gr | |
from transformers import pipeline | |
model = pipeline("image-to-text", model = "Salesforce/blip-image-captioning-base") | |
def describe(raw_image): | |
out = model(raw_image) | |
return out[0]['generated_text'] | |
iface = gr.Interface(describe, | |
inputs = gr.Image(type = 'pil'), | |
outputs = gr.Textbox()) | |
if __name__ == '__main__': | |
iface.launch() | |