emaaaa543's picture
Create app.py
28fb9d0 verified
raw
history blame contribute delete
366 Bytes
from transformers.utils import logging
logging.set_verbosity_error()
import gradio as gr
from transformers import pipeline
pipe= pipeline("image-to-text","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()