From fa57853bd2edab93381f82b4f76c4f4858352284 Mon Sep 17 00:00:00 2001
From: Mert <101130780+mertalev@users.noreply.github.com>
Date: Tue, 8 Aug 2023 13:33:13 -0400
Subject: [PATCH] fix(ml): clear cache if bad zip file (#3600)

---
 machine-learning/app/models/base.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/machine-learning/app/models/base.py b/machine-learning/app/models/base.py
index 8c3a06fc92..9bdbb08b31 100644
--- a/machine-learning/app/models/base.py
+++ b/machine-learning/app/models/base.py
@@ -4,6 +4,7 @@ from abc import ABC, abstractmethod
 from pathlib import Path
 from shutil import rmtree
 from typing import Any
+from zipfile import BadZipFile
 
 from onnxruntime.capi.onnxruntime_pybind11_state import InvalidProtobuf  # type: ignore
 
@@ -23,7 +24,7 @@ class InferenceModel(ABC):
         loader = self.load if eager else self.download
         try:
             loader(**model_kwargs)
-        except (OSError, InvalidProtobuf):
+        except (OSError, InvalidProtobuf, BadZipFile):
             self.clear_cache()
             loader(**model_kwargs)