Update app.py

app.py

@@ -86,13 +86,12 @@ class CustomClient(fl.client.NumPyClient):
         state_dict = OrderedDict({k: torch.Tensor(v) for k, v in params_dict})
         self.net.load_state_dict(state_dict, strict=True)
 
-    def fit(self, parameters, config, round_num, plot_placeholder):
+    def fit(self, parameters, config):
         self.set_parameters(parameters)
         train(self.net, self.trainloader, epochs=1)
         loss, accuracy = test(self.net, self.testloader)
         self.losses.append(loss)
         self.accuracies.append(accuracy)
-        self.plot_metrics(round_num, plot_placeholder)
         return self.get_parameters(config={}), len(self.trainloader.dataset), {}
 
     def evaluate(self, parameters, config):
@@ -144,9 +143,9 @@ def main():
         test_df = pd.DataFrame(test_datasets[i])
 
         st.write("#### Train Dataset")
-        edited_train_df = st.experimental_data_editor(train_df, key=f"train_{i}")
+        edited_train_df = st.data_editor(train_df, key=f"train_{i}")
         st.write("#### Test Dataset")
-        edited_test_df = st.experimental_data_editor(test_df, key=f"test_{i}")
+        edited_test_df = st.data_editor(test_df, key=f"test_{i}")
 
         edited_train_dataset = Dataset.from_pandas(edited_train_df)
         edited_test_dataset = Dataset.from_pandas(edited_test_df)