let hybrid textline light model be loaded

pull/86/head
vahid 1 year ago
parent 29e6ad076f
commit 380f59ad67

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

Loading…
Cancel
Save