Demosthene-OR commited on
Commit
5e30142
1 Parent(s): 1036c97

Update main_dl.py

Browse files
Files changed (1) hide show
  1. main_dl.py +3 -5
main_dl.py CHANGED
@@ -275,9 +275,7 @@ def init_dl_identifier():
275
  merge = Merge(dataPath+"/dl_id_lang_split", dataPath, "dl_tiktoken_id_language_model.h5").merge(cleanup=False)
276
  dl_model = keras.models.load_model(dataPath+"/dl_tiktoken_id_language_model.h5") #, compile=False)
277
  return
278
- '''
279
- init_dl_identifier()
280
- '''
281
  def lang_id_dl(sentences):
282
  global dl_model, label_encoder, lan_to_language
283
 
@@ -332,8 +330,8 @@ async def trad_transformer(lang_tgt:str,
332
  return decode_sequence_transf(texte, "en", "fr")
333
 
334
  @api.get('/small_vocab/plot_model', name="Affiche le modèle")
335
- def affiche_modele(lang_tgt:Optional[str],
336
- model_type: str):
337
  global translation_model, dl_model
338
 
339
  if model_type=="lang_id":
 
275
  merge = Merge(dataPath+"/dl_id_lang_split", dataPath, "dl_tiktoken_id_language_model.h5").merge(cleanup=False)
276
  dl_model = keras.models.load_model(dataPath+"/dl_tiktoken_id_language_model.h5") #, compile=False)
277
  return
278
+
 
 
279
  def lang_id_dl(sentences):
280
  global dl_model, label_encoder, lan_to_language
281
 
 
330
  return decode_sequence_transf(texte, "en", "fr")
331
 
332
  @api.get('/small_vocab/plot_model', name="Affiche le modèle")
333
+ def affiche_modele(model_type: str,
334
+ lang_tgt:Optional[str]=None):
335
  global translation_model, dl_model
336
 
337
  if model_type=="lang_id":