ailm's picture
gradio-application
28fa8d7 verified
raw
history blame
No virus
471 Bytes
import gradio as gr
from transformers import pipeline
model_pipeline = pipeline("image-to-text",
model = "Salesforce/blip-image-captioning-base")
#990M model
def main(input):
output = model_pipeline(input) #globally defined model pipeline
return output[0]['generated_text'] #generated output captoion string
interface_gradio = gr.Interface(
main,
inputs = gr.Image(type = 'pil'),
outputs = "text"
)
interface_gradio.launch()