Update app.py

app.py

@@ -391,7 +391,6 @@
 
 # if __name__ == "__main__":
 #     main()
-
 import streamlit as st
 import matplotlib.pyplot as plt
 import torch
@@ -428,7 +427,7 @@ def load_data(dataset_name, train_size=20, test_size=20, num_clients=2, use_utf8
         tokenized_datasets = tokenized_datasets.rename_column("label", "labels")
     else:
         def utf8_encode_function(examples):
-            examples["text"] = [text.encode('utf-8') for text in examples["text"]]
+            examples["input_ids"] = [list(text.encode('utf-8')) for text in examples["text"]]
             return examples
 
         tokenized_datasets = raw_datasets.map(utf8_encode_function, batched=True)