MJobe commited on
Commit
686167b
1 Parent(s): 42a6b89

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -23,7 +23,7 @@ app.add_middleware(
23
  nlp_qa = pipeline("document-question-answering", model="jinhybr/OCR-DocVQA-Donut")
24
  nlp_qa_v2 = pipeline("document-question-answering", model="faisalraza/layoutlm-invoices")
25
  nlp_qa_v3 = pipeline("question-answering", model="deepset/roberta-base-squad2")
26
- nlp_classification = pipeline("text-classification", model="distilbert-base-uncased-finetuned-sst-2-english")
27
 
28
  description = """
29
  ## Image-based Document QA
 
23
  nlp_qa = pipeline("document-question-answering", model="jinhybr/OCR-DocVQA-Donut")
24
  nlp_qa_v2 = pipeline("document-question-answering", model="faisalraza/layoutlm-invoices")
25
  nlp_qa_v3 = pipeline("question-answering", model="deepset/roberta-base-squad2")
26
+ nlp_classification = pipeline("text-classification", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english")
27
 
28
  description = """
29
  ## Image-based Document QA