feat: add timeout to security issue

Lucs1590 · Nov 13, 2023 · 105e147 · 105e147
1 parent 8a09762
commit 105e147
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/auxiliary.py b/src/auxiliary.py
@@ -628,7 +628,7 @@ def get_image_from_url(url: str) -> np.ndarray:
         np.ndarray: image.
     """
     try:
-        response = requests.get(url)
+        response = requests.get(url, timeout=60)
     except Exception as error:
         raise ConnectionError(
             'you need to be connected to some internet network to download the EAST model.'