locoalien/docsai
0
1from fastapi import FastAPI2from transformers import pipeline3 4nlp = pipeline(5 "document-question-answering",6 model="impira/layoutlm-document-qa",7)8 9app = FastAPI()10 11@app.get("/")12def read_root():13 return nlp(14 "https://templates.invoicehome.com/invoice-template-us-neat-750px.png",15 "What is the invoice number?"16)