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

blakeblackshear · Dec 14, 2024 · 1b7fe95 · 1b7fe95
1 parent 0763f56
commit 1b7fe95
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/frigate/detectors/plugins/deepstack.py b/frigate/detectors/plugins/deepstack.py
@@ -32,6 +32,7 @@ def __init__(self, detector_config: DeepstackDetectorConfig):
         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 @@ def detect_raw(self, tensor_input):
         data = {"api_key": self.api_key}
 
         try:
-            response = requests.post(
+            response = self.session.post(
                 self.api_url,
                 data=data,
                 files={"image": image_bytes},