From 0d00697595f97afa54afe9c30c42a116e3d7846a Mon Sep 17 00:00:00 2001 From: evelynmitchell Date: Sat, 18 Nov 2023 14:19:01 -0700 Subject: [PATCH] 157 possible fix --- swarms/models/layoutlm_document_qa.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/swarms/models/layoutlm_document_qa.py b/swarms/models/layoutlm_document_qa.py index 1688a231..6550518b 100644 --- a/swarms/models/layoutlm_document_qa.py +++ b/swarms/models/layoutlm_document_qa.py @@ -5,7 +5,7 @@ visual question answering on real world docs lik invoice, pdfs, etc from transformers import pipeline -class LayoutLMDocumentQA: +class """ LayoutLMDocumentQA for document question answering: @@ -24,7 +24,6 @@ class LayoutLMDocumentQA: def __init__( self, model_name: str = "impira/layoutlm-document-qa", - task_type: str = "document-question-answering", ): self.model_name = model_name self.task_type = task_type