layoutLM

layoutlm.py

@@ -0,0 +1,12 @@
+from transformers import pipeline
+
+
+def get_result(image_path, question):
+    nlp = pipeline(
+        "document-question-answering",
+        model="impira/layoutlm-document-qa",
+    )
+
+    result = nlp(image_path, question)
+
+    return result[0]['answer']