Change dataset option name

app.py

@@ -6,9 +6,7 @@ import numpy as np
 # Add a title
 st.title('GPT Detection Demo')
 st.write("This is a demo for GPT detection. You can use this demo to test the model. There are 3 variations of the Roberta Classifier Model, The model was trained on CHEAT, GPABenchmark, OpenGPT datasets.You can choose dataset variation of the model on the sidebar.")
-st.write("CHEAT        - Scientific Abstract Generated by GPT3.5")
-st.write("GPABenchmark - Computer Science Abstract Generated by GPT3.5")
-st.write("OpenGPT      - General News Article Generated by GPT3.5")
+st.write("CHEAT        - Scientific Abstract Generated by GPT3.5\nGPABenchmark - Computer Science Abstract Generated by GPT3.5\nOpenGPT      - General News Article Generated by GPT3.5")
 # st.write("Reference on how we built Roberta Sentinel: https://arxiv.org/abs/2305.07969")
 
 # # Add 4 options for 4 models
@@ -18,8 +16,8 @@ st.write("OpenGPT      - General News Article Generated by GPT3.5")
 # )
 
 DatasetOption = st.sidebar.selectbox(
-    'Which Dataset the model was trained on?',
-    ('OpenGPT', 'GPABenchmark', 'CHEAT'),
+    'Select Input Text Domain?',
+    ('General Text', 'Computer Science Abstract', 'Scientific Abstract'),
 )
 
 
@@ -47,13 +45,13 @@ if st.button('Generate'):
     #         result = RobertaClassifierCHEATInference(text)
     #         st.write("Model: RobertaClassifierCHEAT")
 
-    if DatasetOption == 'OpenGPT':
+    if DatasetOption == 'General Text':
         result = RobertaClassifierOpenGPTInference(text)
         st.write("Model: RobertaClassifierOpenGPT")
-    elif DatasetOption == 'GPABenchmark':
+    elif DatasetOption == 'Computer Science Abstract':
         result = RobertaClassifierGPABenchmarkInference(text)
         st.write("Model: RobertaClassifierGPABenchmark")
-    elif DatasetOption == 'CHEAT':
+    elif DatasetOption == 'Scientific Abstract':
         result = RobertaClassifierCHEATInference(text)
         st.write("Model: RobertaClassifierCHEAT")