Rodolfo Torres commited on
Commit
f8e2041
1 Parent(s): dc50f99

Model update

Browse files
Files changed (1) hide show
  1. main.py +1 -2
main.py CHANGED
@@ -13,11 +13,10 @@ import PyPDF2
13
 
14
  print('End loading libraries: ', round(time()-t_ini))
15
  t_ini = time()
16
- model_name = "roaltopo/text-talk-ai_question-answer-12"
17
  qa_pipeline = pipeline(
18
  "question-answering",
19
  model=model_name,
20
- use_auth_token = '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'
21
  )
22
  print('End loading model: ', round(time()-t_ini))
23
  t_ini = time()
 
13
 
14
  print('End loading libraries: ', round(time()-t_ini))
15
  t_ini = time()
16
+ model_name = "roaltopo/scan-u-doc_question-answer"
17
  qa_pipeline = pipeline(
18
  "question-answering",
19
  model=model_name,
 
20
  )
21
  print('End loading model: ', round(time()-t_ini))
22
  t_ini = time()