Update app.py

app.py

@@ -3,11 +3,16 @@ import streamlit as st
 from transformers import pipeline
 
 # Load HuggingFace model pipelines
+# huggingFaceModels = {
+#     "roberta": "Iram7866/roberta_edos_labelled_aggregated",
+#     "fBert": "Iram7866/fbert_edos_labelled_aggregated",
+#     "hateBert": "Iram7866/hatebert_edos_labelled_aggregated",
+#     "berTweet": "Iram7866/bertweet_edos_labelled_aggregated",
+# }
 huggingFaceModels = {
     "roberta": "Iram7866/roberta_edos_labelled_aggregated",
     "fBert": "Iram7866/fbert_edos_labelled_aggregated",
     "hateBert": "Iram7866/hatebert_edos_labelled_aggregated",
-    "berTweet": "Iram7866/bertweet_edos_labelled_aggregated",
 }
 
 # Function to classify text
@@ -34,7 +39,8 @@ st.write("Enter the text you want to classify:")
 textInputBox = st.text_input("Text Goes Here...")
 
 # Model Radio Selection
-selectedModel = st.radio("Select a Model:", ["roberta", "fBert", "hateBert", "berTweet"])
+# selectedModel = st.radio("Select a Model:", ["roberta", "fBert", "hateBert", "berTweet"])
+selectedModel = st.radio("Select a Model:", ["roberta", "fBert", "hateBert"])
 
 labels ={ "LABEL_0": "Not Sexist", "LABEL_1": "Sexist" }