Update app.py

app.py

@@ -1017,6 +1017,15 @@ def read_log_file2():
 
 def main():
     st.write("## Federated Learning with Dynamic Models and Datasets for Mobile Devices")
+    logs = read_log_file2()
+    st.markdown(logs)
+    # Provide a download button for the logs
+    st.download_button(
+        label="Download Logs",
+        data=logs,
+        file_name="./log.txt",
+        mime="text/plain"
+    )
     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"])