let hybrid textline light model be loaded

2 years ago · 380f59ad67
parent 29e6ad076f
commit 380f59ad67
1 changed files with 6 additions and 2 deletions
--- 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:
+            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