From 47737b668a9583c7cc61cf8f9d7642ea632c5a63 Mon Sep 17 00:00:00 2001
From: FL42 <46161216+fl42@users.noreply.github.com>
Date: Sat, 14 Dec 2024 00:17:34 +0100
Subject: [PATCH] fix: use requests.Session() for DeepStack API

---
 frigate/detectors/plugins/deepstack.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/frigate/detectors/plugins/deepstack.py b/frigate/detectors/plugins/deepstack.py
index 20d37fa8e9..e00a4e70d2 100644
--- 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},