Demosthene-OR commited on
Commit
ce16949
·
1 Parent(s): 437f6f5

Update main_dl.py

Browse files
Files changed (1) hide show
  1. main_dl.py +2 -4
main_dl.py CHANGED
@@ -281,8 +281,6 @@ def lang_id_dl(sentences):
281
  global dl_model, label_encoder, lan_to_language
282
 
283
  print("sentences:",sentences)
284
- if (dl_model is not None): print("dl_model OK")
285
- else: print("dl_model vide")
286
  if "str" in str(type(sentences)): predictions = dl_model.predict(encode_text([sentences]))
287
  else: predictions = dl_model.predict(encode_text(sentences))
288
  # Décodage des prédictions en langues
@@ -347,5 +345,5 @@ def affiche_modele(lang_tgt:str,
347
  return Response(content=image_data, media_type="image/png")
348
 
349
  @api.get('/lang_id_dl', name="Id de langue par DL")
350
- def language_id_dl(sentences:List[str] = Query(..., min_length=1)):
351
- return lang_id_dl(sentences)
 
281
  global dl_model, label_encoder, lan_to_language
282
 
283
  print("sentences:",sentences)
 
 
284
  if "str" in str(type(sentences)): predictions = dl_model.predict(encode_text([sentences]))
285
  else: predictions = dl_model.predict(encode_text(sentences))
286
  # Décodage des prédictions en langues
 
345
  return Response(content=image_data, media_type="image/png")
346
 
347
  @api.get('/lang_id_dl', name="Id de langue par DL")
348
+ def language_id_dl(sentence:List[str] = Query(..., min_length=1)):
349
+ return lang_id_dl(sentence)