Update main_dl.py

main_dl.py

@@ -18,6 +18,7 @@ from tensorflow.keras.utils import plot_model
 
 api = FastAPI()
 dataPath = "data"
+imagePath = "images"
 
 # ===== Keras ====
 strip_chars = string.punctuation + "¿"
@@ -287,13 +288,30 @@ def check_api(lang_tgt:str,
     global translation_model
     
     if (lang_tgt=='en'):
-        translation_model = rnn_fr_en
+        translation_model = transformer_fr_en
         return decode_sequence_tranf(texte, "fr", "en")
     else:
-        translation_model = rnn_en_fr
+        translation_model = transformer_en_fr
         return decode_sequence_tranf(texte, "en", "fr")
 
-'''
+@api.get('/small_vocab/plot_model', name="Affiche le modèle")
+def check_api(lang_tgt:str,
+              model_type: str):
+    global translation_model
+    
+    if (lang_tgt=='en'):
+        if model_type=="rnn":
+            translation_model = rnn_fr_en
+        else:
+            translation_model = transformer_fr_en
+    else:
+        if model_type=="rnn":
+            translation_model = rnn_en_fr
+        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')
+    return 
+''' 
 def run():
 
     global n1, df_data_src, df_data_tgt, translation_model, placeholder, model_speech