blip-api / app.py
spuun's picture
Update app.py
c2c8125
raw
history blame
No virus
290 Bytes
import gradio as gr
from transformers import pipeline
model_id = "Salesforce/blip-image-captioning-large"
captioner = pipeline(model=model_id)
def launch(input):
return captioner(input)[0]["generated_text"]
iface = gr.Interface(fn=launch, inputs="pil", outputs="text")
iface.launch()