fix bug and remove model option

app.py

@@ -5,14 +5,14 @@ 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 model, The model was trained on CHEAT, GPABenchmark, OpenGPT datasets. They are all in the domain of Scientific Abstract. You can choose dataset variation of the model on the sidebar.")
+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. They are all in the domain of Scientific Abstract. You can choose dataset variation of the model on the sidebar.")
 # st.write("Reference on how we built Roberta Sentinel: https://arxiv.org/abs/2305.07969")
 
-# Add 4 options for 4 models
-ModelOption = st.sidebar.selectbox(
-    'Which Model do you want to use?',
-    ('RobertaClassifier'),
-)
+# # Add 4 options for 4 models
+# ModelOption = st.sidebar.selectbox(
+#     'Which Model do you want to use?',
+#     ('RobertaClassifier'),
+# )
 
 DatasetOption = st.sidebar.selectbox(
     'Which Dataset the model was trained on?',
@@ -22,6 +22,8 @@ DatasetOption = st.sidebar.selectbox(
 
 text = st.text_area('Enter text here (max 512 words)', '')
 
+
+
 if st.button('Generate'):
     # if ModelOption == 'RobertaSentinel':
     #     if DatasetOption == 'OpenGPT':
@@ -31,16 +33,26 @@ if st.button('Generate'):
     #         result = RobertaSentinelCSAbstractInference(text)
     #         st.write("Model: RobertaSentinelCSAbstract")
 
-    if ModelOption == 'RobertaClassifier':
-        if DatasetOption == 'OpenGPT':
-            result = RobertaClassifierOpenGPTInference(text)
-            st.write("Model: RobertaClassifierOpenGPT")
-        elif DatasetOption == 'GPABenchmark':
-            result = RobertaClassifierGPABenchmarkInference(text)
-            st.write("Model: RobertaClassifierGPABenchmark")
-        elif DatasetOption == 'CHEAT':
-            result = RobertaClassifierCHEATInference(text)
-            st.write("Model: RobertaClassifierCHEAT")
+    # if ModelOption == 'RobertaClassifier':
+    #     if DatasetOption == 'OpenGPT':
+    #         result = RobertaClassifierOpenGPTInference(text)
+    #         st.write("Model: RobertaClassifierOpenGPT")
+    #     elif DatasetOption == 'GPABenchmark':
+    #         result = RobertaClassifierGPABenchmarkInference(text)
+    #         st.write("Model: RobertaClassifierGPABenchmark")
+    #     elif DatasetOption == 'CHEAT':
+    #         result = RobertaClassifierCHEATInference(text)
+    #         st.write("Model: RobertaClassifierCHEAT")
+
+    if DatasetOption == 'OpenGPT':
+        result = RobertaClassifierOpenGPTInference(text)
+        st.write("Model: RobertaClassifierOpenGPT")
+    elif DatasetOption == 'GPABenchmark':
+        result = RobertaClassifierGPABenchmarkInference(text)
+        st.write("Model: RobertaClassifierGPABenchmark")
+    elif DatasetOption == 'CHEAT':
+        result = RobertaClassifierCHEATInference(text)
+        st.write("Model: RobertaClassifierCHEAT")
 
 
     Prediction = "Human Written" if not np.argmax(result) else "Machine Generated"