Update app.py

app.py

@@ -355,8 +355,8 @@ import torch
     
 def main():
     st.write("## Federated Learning with dynamic models and datasets for mobile devices")
-    dataset_name = st.selectbox("Dataset", ["audio_instruction_task","imdb", "amazon_polarity", "ag_news"])
-    model_name = st.selectbox("Model", ["facebook/hubert-base-ls960","bert-base-uncased", "distilbert-base-uncased"])
+    dataset_name = st.selectbox("Dataset", ["imdb","audio_instruction_task", "amazon_polarity", "ag_news"])
+    model_name = st.selectbox("Model", ["bert-base-uncased","facebook/hubert-base-ls960", "distilbert-base-uncased"])
 
     net = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=2).to(DEVICE)
     # processor = Wav2Vec2Processor.from_pretrained(model_name)