Commit
·
5e30142
1
Parent(s):
1036c97
Update main_dl.py
Browse files- 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(
|
336 |
-
|
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":
|