Update modelisation_seq2seq_tab.py

tabs/modelisation_seq2seq_tab.py

@@ -433,7 +433,14 @@ def run():
                 """)
                 , unsafe_allow_html=True)
         st.write("<center><h5>"+tr("Architecture du modèle utilisé")+":</h5>", unsafe_allow_html=True)
-        plot_model(translation_model, show_shapes=True, show_layer_names=True, show_layer_activations=True,rankdir='TB',to_file=st.session_state.ImagePath+'/model_plot.png')
+        # URL de votre endpoint FastAPI avec les paramètres de requête
+        url = "https://demosthene-or-api-avr23-cds-translation.hf.space/small_vocab/plot_model"
+        if (chosen_id == "tab1"):
+            params = {"lang_tgt": Lang[-2:], "model_type": "rnn"}
+        else:
+            params = {"lang_tgt": Lang[-2:], "model_type": "transformer"}
+        # Envoie d'une requête GET avec les paramètres de requête
+        response = requests.get(url, params=params)    
         st.image(st.session_state.ImagePath+'/model_plot.png',use_column_width=True)
         st.write("</center>", unsafe_allow_html=True)