Demosthene-OR commited on
Commit
437f6f5
·
1 Parent(s): dc17887

Update main_dl.py

Browse files
Files changed (1) hide show
  1. main_dl.py +4 -1
main_dl.py CHANGED
@@ -271,7 +271,10 @@ def init_dl_identifier():
271
  lan_identified = [lan_to_language[l] for l in list_lan]
272
  label_encoder.fit(list_lan)
273
  merge = Merge(dataPath+"/dl_id_lang_split", dataPath, "dl_tiktoken_id_language_model.h5").merge(cleanup=False)
274
- dl_model = keras.models.load_model(dataPath+"/dl_tiktoken_id_language_model.h5")
 
 
 
275
  return
276
 
277
  def lang_id_dl(sentences):
 
271
  lan_identified = [lan_to_language[l] for l in list_lan]
272
  label_encoder.fit(list_lan)
273
  merge = Merge(dataPath+"/dl_id_lang_split", dataPath, "dl_tiktoken_id_language_model.h5").merge(cleanup=False)
274
+ dl_model = keras.models.load_model(dataPath+"/dl_tiktoken_id_language_model.h5", compile=False)
275
+
276
+ if (dl_model is not None): print("dl_model OK")
277
+ else: print("dl_model vide")
278
  return
279
 
280
  def lang_id_dl(sentences):