File size: 408 Bytes
c62b051
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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

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()