que_ans / app.py
786avinash's picture
Update app.py
98d3845 verified
raw
history blame
380 Bytes
from transformers import BlipForQuestionAnswering
from transformers import AutoProcessor
from PIL import Image
import gradio as gr
def qna(image,question):
inputs=processor(image,question,return_tensors="pt")
out=model.generate(**inputs)
return processor.decode(out[0],skip_special_tokens=True)
interf=gr.Interface(qna,inputs=["image","text"],outputs="text")
interf.launch()