Update app.py

app.py

@@ -471,10 +471,8 @@ def plot_metrics(rounds, clients, memory_usage):
 def read_log_file2():
     with open("./log.txt", "r") as file:
         return file.read()
-import streamlit as st
 def main():
-    
-    st.markdown(print(st.logger._loggers))
+
     st.write("## Federated Learning with Dynamic Models and Datasets for Mobile Devices")
     dataset_name = st.selectbox("Dataset", ["imdb", "amazon_polarity", "ag_news"])
     model_name = st.selectbox("Model", ["bert-base-uncased", "facebook/hubert-base-ls960", "distilbert-base-uncased"])
@@ -534,7 +532,9 @@ def main():
         )
 
         for round_num in range(NUM_ROUNDS):
-            st.write(f"### Round {round_num + 1}")
+            st.write(f"### Round {round_num + 1} ✅")
+
+            st.markdown(print(st.logger._loggers))
             st.markdown(read_log_file2())
             logs = read_log_file2()
             import re