fixed return value

OpenAdaptAI · FFFiend · Jun 23, 2023 · Jun 23, 2023 · Jun 23, 2023 · Jun 23, 2023
commit 3ea5cf1b135473a9b05cd23f5782ef4d4746df50
diff --git a/openadapt/strategies/mixins/layout.py b/openadapt/strategies/mixins/layout.py
@@ -36,7 +36,7 @@ def document_query(
         query_pipeline = pipeline("document-question-answering",
             model = "impira/layoutlm-document-qa")
 
-        return query_pipeline(image, question)['answer']
+        return query_pipeline(image, question)[0]['answer']
 
     def extract_table(
         self