diff --git a/qurator/eynollah/eynollah.py b/qurator/eynollah/eynollah.py index 1346672..4fecfed 100644 --- a/qurator/eynollah/eynollah.py +++ b/qurator/eynollah/eynollah.py @@ -665,8 +665,12 @@ class Eynollah: if model_dir in self.models: model = self.models[model_dir] else: - model = load_model(model_dir, compile=False) - self.models[model_dir] = model + try: + model = load_model(model_dir, compile=False) + self.models[model_dir] = model + except: + model = load_model(model_dir , compile=False,custom_objects = {"PatchEncoder": PatchEncoder, "Patches": Patches}) + self.models[model_dir] = model return model, None