Update app.py

app.py

@@ -358,7 +358,7 @@ def main():
     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"])
 
-    # net = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=2).to(DEVICE)
+    net = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=2).to(DEVICE)
     # processor = Wav2Vec2Processor.from_pretrained(model_name)
     # net = HubertForSequenceClassification.from_pretrained(model_name, num_labels=2).to(DEVICE)