File size: 365 Bytes
56a0e24
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

import os
import gradio as gr
from transformers import pipeline

model_id = "Salesforce/blip-vqa-base"

pipe = pipeline("image-to-text", model=model_id)

def launch(input):
    out = pipe(input)
    return out[0]['generated_text']

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

iface.launch()