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":
|