fix: use requests.Session() for DeepStack API (#15505)

This commit is contained in:
FL42 2024-12-14 15:54:13 +01:00 committed by GitHub
parent 0763f56047
commit 1b7fe9523d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -32,6 +32,7 @@ class DeepStack(DetectionApi):
self.api_timeout = detector_config.api_timeout
self.api_key = detector_config.api_key
self.labels = detector_config.model.merged_labelmap
self.session = requests.Session()
def get_label_index(self, label_value):
if label_value.lower() == "truck":
@ -51,7 +52,7 @@ class DeepStack(DetectionApi):
data = {"api_key": self.api_key}
try:
response = requests.post(
response = self.session.post(
self.api_url,
data=data,
files={"image": image_bytes},