Update app.py

app.py

@@ -18,11 +18,9 @@ NUM_LAYERS = 2 # Number of transformer layers
 
 # Gumbel softmax temperature
 TAU = 1.0
-
 LEARNING_RATE = 1e-3
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-torch.random.manual_seed(1024)
 
 # Pass embeded into decoder instead of using the original x
 class TransformerEncoder(nn.Module):
@@ -266,7 +264,7 @@ def run_signal_game(NUM_SENDERS, NUM_RECEIVERS, num_rounds):
     st.pyplot(fig)
 
 # Streamlit UI
-st.title('Multi-Agents Signal Game')
+st.title('Multi-Agent Signal Game')
 
 NUM_SENDERS = st.sidebar.slider("NUM_SENDERS", 1, 3, 2)
 NUM_RECEIVERS = st.sidebar.slider("NUM_RECEIVERS", 1, 3, 2)