MJobe commited on
Commit
57d4ce7
1 Parent(s): 0c14995

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -10,7 +10,7 @@ from starlette.middleware.cors import CORSMiddleware
10
  app = FastAPI()
11
 
12
  # Use a pipeline as a high-level helper
13
- nlp_qa = pipeline("document-question-answering", model="impira/layoutlm-document-qa")
14
  # Use a pipeline as a high-level helper
15
  nlp_ner = pipeline('question-answering', model='deepset/roberta-base-squad2', tokenizer='deepset/roberta-base-squad2')
16
 
 
10
  app = FastAPI()
11
 
12
  # Use a pipeline as a high-level helper
13
+ nlp_qa = pipeline("document-question-answering", model="naver-clova-ix/donut-base-finetuned-docvqa")
14
  # Use a pipeline as a high-level helper
15
  nlp_ner = pipeline('question-answering', model='deepset/roberta-base-squad2', tokenizer='deepset/roberta-base-squad2')
16