Update app.py

app.py

@@ -270,7 +270,7 @@ def get_splade_doc_representation(text):
 
     sorted_representation = sorted(meaningful_tokens.items(), key=lambda item: item[0]) # Sort alphabetically for clarity
 
-    formatted_output = "Binary Bag-of-Words Representation:\n\n"
+    formatted_output = "Binary:\n\n"
     if not sorted_representation:
         formatted_output += "No significant terms found for this input.\n"
     else:
@@ -292,7 +292,7 @@ def predict_representation_explorer(model_choice, text):
         return get_splade_cocondenser_representation(text)
     elif model_choice == "MLP encoder (SPLADE-v3-lexical)":
         return get_splade_lexical_representation(text)
-    elif model_choice == "Binary Bag-of-Words": # Changed name
+    elif model_choice == "Binary": # Changed name
         return get_splade_doc_representation(text)
     else:
         return "Please select a model.", "" # Return two empty strings for consistency
@@ -414,7 +414,7 @@ def get_model_assets(model_choice_str):
         return get_splade_cocondenser_vector, tokenizer_splade, False, "MLM encoder (SPLADE-cocondenser-distil)"
     elif model_choice_str == "MLP encoder (SPLADE-v3-lexical)":
         return get_splade_lexical_vector, tokenizer_splade_lexical, False, "MLP encoder (SPLADE-v3-lexical)"
-    elif model_choice_str == "Binary Bag-of-Words":
+    elif model_choice_str == "Binary":
         return get_splade_doc_vector, tokenizer_splade_doc, True, "Binary Bag-of-Words"
     else:
         return None, None, False, "Unknown Model"
@@ -480,7 +480,7 @@ with gr.Blocks(title="SPLADE Demos", css=css) as demo:
                         [
                             "MLM encoder (SPLADE-cocondenser-distil)",
                             "MLP encoder (SPLADE-v3-lexical)",
-                            "Binary Bag-of-Words"
+                            "Binary"
                         ],
                         label="Choose Sparse Encoder",
                         value="MLM encoder (SPLADE-cocondenser-distil)"
@@ -549,7 +549,7 @@ with gr.Blocks(title="SPLADE Demos", css=css) as demo:
             model_choices = [
                 "MLM encoder (SPLADE-cocondenser-distil)",
                 "MLP encoder (SPLADE-v3-lexical)",
-                "Binary Bag-of-Words"
+                "Binary"
             ]
 
             # Input components for the second tab