pstayal's picture
Update app.py
2efe7bf verified
Raw History Blame Contribute Delete
442 Bytes
import gradio as gr
from transformers import pipeline
pipe = pipeline("visual-question-answering", model = "openbmb/MiniCPM-Llama3-V-2_5", trust_remote_code=True)
#model="Salesforce/blip2-opt-2.7b")
def launch(input1, input2):
out = pipe(input1, input2)
return out#['generated_text']
iface = gr.Interface(launch,
inputs=[gr.Image(type='pil'), gr.Textbox()],
outputs="text")
iface.launch()