Spaces:
Sleeping
Sleeping
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()
|