Added a few qol changes

app.py

@@ -20,6 +20,25 @@ def update_textboxes(n_visible):
             updates.append(gr.update(visible=False))
     return n_visible, *updates
 
+# Function to decrease visible textboxes
+def remove_textboxes(n_visible):
+    """
+    Given the current visible count, decrements it by 1 (down to 2)
+    and returns updated visibility settings for all model textboxes,
+    clearing the content of any hidden ones.
+    """
+    if n_visible > 2:
+        n_visible -= 1
+    updates = []
+    for i in range(MAX_MODELS):
+        if i < n_visible:
+            # keep visible, preserve existing value
+            updates.append(gr.update(visible=True))
+        else:
+            # hide and clear content
+            updates.append(gr.update(visible=False, value=""))
+    return (n_visible, *updates)
+
 def run_scoring(input_text, model1, model2, model3, model4, model5, model6, model7, model8, model9, model10, threshold_choice, custom_threshold):
     """
     Collect all non-empty model paths, instantiate Mosaic, compute the score,
@@ -46,7 +65,7 @@ def run_scoring(input_text, model1, model2, model3, model4, model5, model6, mode
     if final_score < threshold:
         result_message = "This text was probably generated."
     else:
-        result_message = "This text is likely human-generated."
+        result_message = "This text is likely human-written."
     return result_message, final_score, threshold
 
 with gr.Blocks() as demo:
@@ -54,29 +73,37 @@ with gr.Blocks() as demo:
     with gr.Row():
         input_text = gr.Textbox(lines=10, placeholder="Enter text here...", label="Input Text")
     with gr.Column():
+        gr.Markdown("**⚠️ Please make sure all models have the same tokenizer or it won’t work.**")
         gr.Markdown("### Model Paths (at least 2 required)")
         gr.Markdown("Order matters for model 1 only, the Reference model. Please use the one with the best perplexity on human texts. (The largest LLM if applicable.) GPT2 models are enough to detect easy prompts from chatgpt.")
         # State to keep track of the number of visible textboxes (starting with 2)
-        n_models_state = gr.State(2)
+        n_models_state = gr.State(4)
         # Create 10 textboxes. We'll name them model1, model2, ..., model10.
-        model1 = gr.Textbox(value="openai-community/gpt2-large", label="Model 1 Path ", visible=True)
-        model2 = gr.Textbox(value="openai-community/gpt2-medium", label="Model 2 Path", visible=True)
-        model3 = gr.Textbox(value="", label="Model 3 Path", visible=False)
-        model4 = gr.Textbox(value="", label="Model 4 Path", visible=False)
+        model1 = gr.Textbox(value="Unbabel/TowerBase-13B-v0.1", label="Model 1 Path ", visible=True)
+        model2 = gr.Textbox(value="Unbabel/TowerBase-7B-v0.1", label="Model 2 Path", visible=True)
+        model3 = gr.Textbox(value="meta-llama/Llama-2-7b-chat-hf", label="Model 3 Path", visible=True)
+        model4 = gr.Textbox(value="meta-llama/Llama-2-7b-hf", label="Model 4 Path", visible=True)
         model5 = gr.Textbox(value="", label="Model 5 Path", visible=False)
         model6 = gr.Textbox(value="", label="Model 6 Path", visible=False)
         model7 = gr.Textbox(value="", label="Model 7 Path", visible=False)
         model8 = gr.Textbox(value="", label="Model 8 Path", visible=False)
         model9 = gr.Textbox(value="", label="Model 9 Path", visible=False)
         model10 = gr.Textbox(value="", label="Model 10 Path", visible=False)
-        # Add a plus button to reveal one more textbox.
-        plus_button = gr.Button("+", elem_id="plus_button")
-        # When plus_button is clicked, update n_models_state and all model textboxes.
+        # Add plus and minus buttons to reveal/hide textboxes.
+        with gr.Row():
+            plus_button = gr.Button("+", elem_id="plus_button")
+            minus_button = gr.Button("-", elem_id="minus_button")
+        # Bind click events
         plus_button.click(
             fn=update_textboxes,
             inputs=n_models_state,
             outputs=[n_models_state, model1, model2, model3, model4, model5, model6, model7, model8, model9, model10]
         )
+        minus_button.click(
+            fn=remove_textboxes,
+            inputs=n_models_state,
+            outputs=[n_models_state, model1, model2, model3, model4, model5, model6, model7, model8, model9, model10]
+        )
     with gr.Row():
         threshold_choice = gr.Radio(choices=["default", "raid", "custom"], value="default", label="Threshold Choice")
         custom_threshold = gr.Number(value=0.0, label="Custom Threshold (if 'custom' selected)")