Update main_dl.py

main_dl.py

@@ -335,19 +335,21 @@ async def trad_transformer(lang_tgt:str,
 @api.get('/small_vocab/plot_model', name="Affiche le modèle")
 def affiche_modele(lang_tgt:str,
               model_type: str):
-    global translation_model
+    global translation_model, dl_model
     
-    if (lang_tgt=='en'):
-        if model_type=="rnn":
-            translation_model = rnn_fr_en
+    if model_type=="lang_id":
+        model_to_display = dl_model
+    elif (model_type=="rnn"):
+        if (lang_tgt=='en'):
+            model_to_display = rnn_fr_en
         else:
-            translation_model = transformer_fr_en
+            model_to_display = rnn_en_fr
     else:
-        if model_type=="rnn":
-            translation_model = rnn_en_fr
+        if (lang_tgt=='en'):
+            model_to_display = transformer_fr_en
         else:
-            translation_model = transformer_en_fr
-    plot_model(translation_model, show_shapes=True, show_layer_names=True, show_layer_activations=True,rankdir='TB',to_file=imagePath+'/model_plot.png')
+            model_to_display = transformer_en_fr
+    plot_model(model_to_display, show_shapes=True, show_layer_names=True, show_layer_activations=True,rankdir='TB',to_file=imagePath+'/model_plot.png')
     with open(imagePath+'/model_plot.png', "rb") as image_file:
             # Lire les données de l'image
             image_data = image_file.read()