LM77's picture
Update app.py
c5b6365 verified
raw
history blame contribute delete
412 Bytes
from transformers.utils import logging
logging.set_verbosity_error()
import gradio as gr
from transformers import pipeline
pipe = pipeline('image-to-text', model='Salesforce/blip-image-captioning-base')
def launch(input):
out = pipe(input)
return out[0]['generated_text']
iface = gr.Interface(launch,
inputs=gr.Image(type='pil'),
outputs="text")
iface.launch()