Demosthene-OR commited on
Commit
9470f04
·
1 Parent(s): b9774ad

Update main_dl.py

Browse files
Files changed (1) hide show
  1. main_dl.py +6 -4
main_dl.py CHANGED
@@ -277,10 +277,11 @@ def check_api(lang_tgt:str,
277
  print("texte: ",texte)
278
  if (lang_tgt=='en'):
279
  translation_model = rnn_fr_en
280
- return decode_sequence_rnn(texte, "en", "fr")
 
281
  else:
282
  translation_model = rnn_en_fr
283
- return decode_sequence_rnn(texte, "fr", "en")
284
 
285
  @api.get('/small_vocab/transformer', name="Traduction par Transformer")
286
  def check_api(lang_tgt:str,
@@ -289,10 +290,11 @@ def check_api(lang_tgt:str,
289
 
290
  if (lang_tgt=='en'):
291
  translation_model = rnn_fr_en
292
- return decode_sequence_tranf(texte, "en", "fr")
 
293
  else:
294
  translation_model = rnn_en_fr
295
- return decode_sequence_tranf(texte, "fr", "en")
296
 
297
  '''
298
  def run():
 
277
  print("texte: ",texte)
278
  if (lang_tgt=='en'):
279
  translation_model = rnn_fr_en
280
+ texte = "california est généralement calme en mars et il est généralement chaud en juin"
281
+ return decode_sequence_rnn(texte, "fr", "en")
282
  else:
283
  translation_model = rnn_en_fr
284
+ return decode_sequence_rnn(texte, "en", "fr")
285
 
286
  @api.get('/small_vocab/transformer', name="Traduction par Transformer")
287
  def check_api(lang_tgt:str,
 
290
 
291
  if (lang_tgt=='en'):
292
  translation_model = rnn_fr_en
293
+ texte = "california est généralement calme en mars et il est généralement chaud en juin"
294
+ return decode_sequence_tranf(texte, "fr", "en")
295
  else:
296
  translation_model = rnn_en_fr
297
+ return decode_sequence_tranf(texte, "en", "fr")
298
 
299
  '''
300
  def run():