fatty-belly commited on
Commit
718c5cc
·
verified ·
1 Parent(s): f7a0ff3

Increase read timeout and max retries in API call

Browse files
pipelines/vqa_extract_optimized_pipeline.py CHANGED
@@ -30,6 +30,8 @@ class PDF_VQA_extract_optimized_pipeline(PipelineABC):
30
  key_name_of_api_key="DF_API_KEY",
31
  model_name=model_name,
32
  max_workers=max_workers,
 
 
33
  )
34
 
35
  self.vqa_extract_prompt = QAExtractPrompt()
 
30
  key_name_of_api_key="DF_API_KEY",
31
  model_name=model_name,
32
  max_workers=max_workers,
33
+ read_timeout=600.0,
34
+ max_retries=2
35
  )
36
 
37
  self.vqa_extract_prompt = QAExtractPrompt()