| import gradio as gr | |
| from transformers import pipeline | |
| nlp = pipeline( | |
| "document-question-answering", | |
| model="impira/layoutlm-document-qa", | |
| ) | |
| def predict(img, text): | |
| prediction = nlp( | |
| img, | |
| text | |
| ) | |
| return prediction[0]['answer'] | |
| gr.Interface(predict, [gr.Image(type="filepath"), gr.Textbox(label="Question")], gr.Textbox()).launch() |