let hybrid textline light model be loaded

pull/86/head
vahid 2 years ago
parent 29e6ad076f
commit 380f59ad67

@ -665,8 +665,12 @@ class Eynollah:
if model_dir in self.models: if model_dir in self.models:
model = self.models[model_dir] model = self.models[model_dir]
else: else:
model = load_model(model_dir, compile=False) try:
self.models[model_dir] = model 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 return model, None

Loading…
Cancel
Save