Demosthene-OR
commited on
Commit
•
cb82670
1
Parent(s):
5e28e89
Update main_dl.py
Browse files- main_dl.py +3 -3
main_dl.py
CHANGED
@@ -271,7 +271,7 @@ def check_api():
|
|
271 |
return {'message': "L'API fonctionne"}
|
272 |
|
273 |
@api.get('/small_vocab/rnn', name="Traduction par RNN")
|
274 |
-
def
|
275 |
texte: str):
|
276 |
global translation_model
|
277 |
|
@@ -283,7 +283,7 @@ def check_api(lang_tgt:str,
|
|
283 |
return decode_sequence_rnn(texte, "en", "fr")
|
284 |
|
285 |
@api.get('/small_vocab/transformer', name="Traduction par Transformer")
|
286 |
-
def
|
287 |
texte: str):
|
288 |
global translation_model
|
289 |
|
@@ -295,7 +295,7 @@ def check_api(lang_tgt:str,
|
|
295 |
return decode_sequence_tranf(texte, "en", "fr")
|
296 |
|
297 |
@api.get('/small_vocab/plot_model', name="Affiche le modèle")
|
298 |
-
def
|
299 |
model_type: str):
|
300 |
global translation_model
|
301 |
|
|
|
271 |
return {'message': "L'API fonctionne"}
|
272 |
|
273 |
@api.get('/small_vocab/rnn', name="Traduction par RNN")
|
274 |
+
def trad_rnn(lang_tgt:str,
|
275 |
texte: str):
|
276 |
global translation_model
|
277 |
|
|
|
283 |
return decode_sequence_rnn(texte, "en", "fr")
|
284 |
|
285 |
@api.get('/small_vocab/transformer', name="Traduction par Transformer")
|
286 |
+
def trad_transformer(lang_tgt:str,
|
287 |
texte: str):
|
288 |
global translation_model
|
289 |
|
|
|
295 |
return decode_sequence_tranf(texte, "en", "fr")
|
296 |
|
297 |
@api.get('/small_vocab/plot_model', name="Affiche le modèle")
|
298 |
+
def affiche_modele(lang_tgt:str,
|
299 |
model_type: str):
|
300 |
global translation_model
|
301 |
|