let hybrid textline light model be loaded

This commit is contained in:
vahid 2023-04-18 15:06:18 +02:00
parent 29e6ad076f
commit 380f59ad67

View file

@ -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