Update app.py

app.py

@@ -125,7 +125,7 @@ class CustomClient(fl.client.NumPyClient):
 def main():
     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", "distilbert-base-uncased"])
+    model_name = st.selectbox("Model", ["bert-base-uncased","facebook/hubert-base-ls960", "distilbert-base-uncased"])
 
     NUM_CLIENTS = st.slider("Number of Clients", min_value=1, max_value=10, value=2)
     NUM_ROUNDS = st.slider("Number of Rounds", min_value=1, max_value=10, value=3)