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