MJobe commited on
Commit
0e96068
1 Parent(s): 68fee4f

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -33,7 +33,7 @@ app.add_middleware(
33
  )
34
 
35
  nlp_qa = pipeline("document-question-answering", model="jinhybr/OCR-DocVQA-Donut")
36
- nlp_qa_v2 = pipeline("document-question-answering", model="faisalraza/layoutlm-invoices")
37
  nlp_qa_v3 = pipeline("question-answering", model="deepset/roberta-base-squad2")
38
  nlp_classification = pipeline("text-classification", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english")
39
  nlp_classification_v2 = pipeline("text-classification", model="cardiffnlp/twitter-roberta-base-sentiment-latest")
 
33
  )
34
 
35
  nlp_qa = pipeline("document-question-answering", model="jinhybr/OCR-DocVQA-Donut")
36
+ nlp_qa_v2 = pipeline("document-question-answering", model="faisalraza/layoutlm-invoices", ignore_mismatched_sizes=True)
37
  nlp_qa_v3 = pipeline("question-answering", model="deepset/roberta-base-squad2")
38
  nlp_classification = pipeline("text-classification", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english")
39
  nlp_classification_v2 = pipeline("text-classification", model="cardiffnlp/twitter-roberta-base-sentiment-latest")