Updated models

app.py

@@ -193,7 +193,7 @@ def predict_mc(model, tokenizer, text):
     return mc_score
 
 
-def ai_generated_test(input, models):
+def ai_generated_test(input):
 
     cleaned_text = remove_special_characters(input)
     bc_score = predict_bc(text_bc_model, text_bc_tokenizer, cleaned_text)
@@ -209,7 +209,7 @@ def ai_generated_test(input, models):
 # COMBINED
 def main(
     input,
-    models,
+    # models,
     year_from,
     month_from,
     day_from,
@@ -218,7 +218,7 @@ def main(
     day_to,
     domains_to_skip,
 ):
-    bc_score, mc_score = ai_generated_test(input, models)
+    bc_score, mc_score = ai_generated_test(input)
     formatted_tokens = plagiarism_check(
         input,
         year_from,
@@ -287,12 +287,12 @@ with gr.Blocks() as demo:
         """
     )
     
-    models = gr.Dropdown(
-            model_list,
-            value=model_list,
-            multiselect=True,
-            label="Models to test against",
-        )
+    # models = gr.Dropdown(
+            # model_list,
+            # value=model_list,
+            # multiselect=True,
+            # label="Models to test against",
+        # )
     
     with gr.Row():
         with gr.Column():
@@ -345,7 +345,7 @@ with gr.Blocks() as demo:
         fn=main,
         inputs=[
             input_text,
-            models,
+            # models,
             year_from,
             month_from,
             day_from,
@@ -364,7 +364,7 @@ with gr.Blocks() as demo:
 
     only_ai_btn.click(
         fn=ai_generated_test,
-        inputs=[input_text, models],
+        inputs=[input_text],
         outputs=[
             bcLabel,
             mcLabel,