Update app.py

app.py

@@ -391,6 +391,10 @@ def format_sparse_vector_output(splade_vector, tokenizer, is_binary=False):
     else:
         terms_list = []
         for i, (term, weight) in enumerate(sorted_representation):
+            # Limiting to 50 terms for display to avoid overly long output
+            if i >= 50:
+                terms_list.append(f"...and {len(sorted_representation) - 50} more terms.")
+                break
             if is_binary:
                 terms_list.append(f"**{term}**")
             else:
@@ -579,6 +583,22 @@ with gr.Blocks(title="SPLADE Demos", css=css) as demo:
                 outputs=gr.Markdown(),
                 allow_flagging="never"
             )
+    
+    # --- ADDED CITATION HERE ---
+    gr.Markdown(
+        """
+        ---
+        ### References
+
+        This demo utilizes **SPLADE-v3** models. For more details, please refer to the original paper:
+
+        * **SPLADE-v3: New baselines for SPLADE**
+            * Carlos Lassance, Hervé Déjean, Thibault Formal, Stéphane Clinchant
+            * arXiv preprint: [2403.06789](https://arxiv.org/abs/2403.06789) (2024)
+            * Licensed under Creative Commons Attribution Non Commercial Share Alike 4.0 International.
+        """
+    )
+
 
 # This block ensures the share URL is captured when the app launches
 if __name__ == "__main__":