qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-07-18 23:39:54 +02:00
Code Issues Projects Releases Wiki Activity

let hybrid textline light model be loaded

Browse source
This commit is contained in:
vahid 2023-04-18 15:06:18 +02:00
parent 29e6ad076f
commit 380f59ad67
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
qurator/eynollah/eynollah.py
View file

@ -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:
try:
model = load_model(model_dir, compile=False) model = load_model(model_dir, compile=False)
self.models[model_dir] = model 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