Update app.py

app.py

@@ -143,9 +143,6 @@ val_dataloader = DataLoader(val_dataset, batch_size=BATCH_SIZE, shuffle=True)
 
 VOCAB_SIZE = len(vocab)
 
-
-
-
 class MultiMultiSignalingGame:
     def __init__(self, senders: list, receivers: list, optimizer, criterion):
         self.senders = senders
@@ -230,8 +227,8 @@ def train_signal_game(NUM_SENDERS, NUM_RECEIVERS, num_rounds):
     output_sentences = []
 
     # Use Streamlit's progress bar
-    progress_bar = st.progress(0)
     loss_plot_placeholder = st.empty()  # 创建一个空位占位符来显示损失图
+    progress_bar = st.progress(0)
     interactions_placeholder = st.empty()  # 创建一个空位占位符来显示交互
 
     for round in range(num_rounds):
@@ -249,13 +246,12 @@ def train_signal_game(NUM_SENDERS, NUM_RECEIVERS, num_rounds):
         ax.set_ylabel('Loss')
         ax.legend()
         loss_plot_placeholder.pyplot(fig)
+        progress_bar.progress(round / num_rounds)
         # 刷新显示每次交互的句子
         interaction_str = "\n\n".join([f"Sender {i} -> Receiver {j}\nSend(encode): {input_sentence}\nReceive(decode): {output_sentence}" 
                                        for i, j, input_sentence, output_sentence in interactions])
         interactions_placeholder.text(interaction_str)
 
-        progress_bar.progress(round / num_rounds)
-
     # Dynamic plotting of the losses
     fig, ax = plt.subplots()
     ax.plot(losses, label='Total Losses', color='blue')