Update app.py
Browse files
app.py
CHANGED
@@ -95,7 +95,7 @@ class CustomClient(fl.client.NumPyClient):
|
|
95 |
return float(loss), len(self.testloader.dataset), {"accuracy": float(accuracy)}
|
96 |
|
97 |
def main():
|
98 |
-
st.write("## Federated Learning with Dynamic Models and Datasets for Mobile Devices")
|
99 |
dataset_name = st.selectbox("Dataset", ["imdb", "amazon_polarity", "ag_news"])
|
100 |
model_name = st.selectbox("Model", ["bert-base-uncased", "distilbert-base-uncased"])
|
101 |
|
|
|
95 |
return float(loss), len(self.testloader.dataset), {"accuracy": float(accuracy)}
|
96 |
|
97 |
def main():
|
98 |
+
st.write("## Federated Learning with Flower and Dynamic Models and Datasets for Mobile Devices")
|
99 |
dataset_name = st.selectbox("Dataset", ["imdb", "amazon_polarity", "ag_news"])
|
100 |
model_name = st.selectbox("Model", ["bert-base-uncased", "distilbert-base-uncased"])
|
101 |
|