Update main_dl.py

main_dl.py

@@ -283,12 +283,11 @@ def lang_id_dl(sentences):
     
     if 'dl_model' not in globals():
         init_dl_identifier()
-    if "str" in str(type(sentences)): predictions = dl_model.predict(encode_text([sentences]))
-    else:  predictions = dl_model.predict(encode_text(sentences))
+    predictions = dl_model.predict(encode_text(sentences))
     # Décodage des prédictions en langues
     predicted_labels_encoded = np.argmax(predictions, axis=1)
     predicted_languages = label_encoder.classes_[predicted_labels_encoded]
-    if "str" in str(type(sentences)): return lan_to_language[predicted_languages[0]]
+    if (len(sentences)==1): return lan_to_language[predicted_languages[0]]
     else: return [l for l in predicted_languages]
 
 #  ==== Endpoints ====
@@ -333,8 +332,8 @@ async def trad_transformer(lang_tgt:str,
         return decode_sequence_transf(texte, "en", "fr")
 
 @api.get('/small_vocab/plot_model', name="Affiche le modèle")
-def affiche_modele(lang_tgt:str,
-              model_type: str):
+def affiche_modele(lang_tgt:Optional[str],
+                   model_type: str):
     global translation_model, dl_model
     
     if model_type=="lang_id":