Added Inseq citation

app.py

@@ -194,7 +194,7 @@ with gr.Blocks(css=custom_css) as demo:
         )
     with gr.Tab("⚙️ Parameters") as params_tab:
         gr.Markdown(
-            "## ✨ Presets\nSelect a preset to load default parameters into the fields below."
+            "## ✨ Presets\nSelect a preset to load default parameters into the fields below. ⚠️ This will overwrite existing parameters."
         )
         with gr.Row(equal_height=True):
             with gr.Column():
@@ -453,7 +453,7 @@ with gr.Blocks(css=custom_css) as demo:
         ],
     )
 
-    load_model_button.click(
+    load_model_event = load_model_button.click(
         preload_model,
         inputs=load_model_args,
         outputs=[],
@@ -481,12 +481,12 @@ with gr.Blocks(css=custom_css) as demo:
 
     # Presets
 
-    default_preset.click(**reset_kwargs).success(preload_model, inputs=load_model_args)
+    default_preset.click(**reset_kwargs).success(preload_model, inputs=load_model_args, cancels=load_model_event)
 
     cora_preset.click(**reset_kwargs).then(
         set_cora_preset,
         outputs=[model_name_or_path, input_template, contextless_input_current_text],
-    ).success(preload_model, inputs=load_model_args)
+    ).success(preload_model, inputs=load_model_args, cancels=load_model_event)
 
     zephyr_preset.click(**reset_kwargs).then(
         set_zephyr_preset,
@@ -496,12 +496,12 @@ with gr.Blocks(css=custom_css) as demo:
             contextless_input_current_text,
             decoder_input_output_separator,
         ],
-    ).success(preload_model, inputs=load_model_args)
+    ).success(preload_model, inputs=load_model_args, cancels=load_model_event)
 
     multilingual_mt_template.click(**reset_kwargs).then(
         set_mmt_preset,
         outputs=[model_name_or_path, input_template, output_template, tokenizer_kwargs],
-    ).success(preload_model, inputs=load_model_args)
+    ).success(preload_model, inputs=load_model_args, cancels=load_model_event)
 
     chatml_template.click(**reset_kwargs).then(
         set_chatml_preset,
@@ -512,7 +512,7 @@ with gr.Blocks(css=custom_css) as demo:
             decoder_input_output_separator,
             special_tokens_to_keep,
         ],
-    ).success(preload_model, inputs=load_model_args)
+    ).success(preload_model, inputs=load_model_args, cancels=load_model_event)
 
     towerinstruct_template.click(**reset_kwargs).then(
         set_towerinstruct_preset,
@@ -523,7 +523,7 @@ with gr.Blocks(css=custom_css) as demo:
             decoder_input_output_separator,
             special_tokens_to_keep,
         ],
-    ).success(preload_model, inputs=load_model_args)
+    ).success(preload_model, inputs=load_model_args, cancels=load_model_event)
 
     gemma_template.click(**reset_kwargs).then(
         set_gemma_preset,
@@ -534,6 +534,6 @@ with gr.Blocks(css=custom_css) as demo:
             decoder_input_output_separator,
             special_tokens_to_keep,
         ],
-    ).success(preload_model, inputs=load_model_args)
+    ).success(preload_model, inputs=load_model_args, cancels=load_model_event)
 
 demo.launch(allowed_paths=["outputs/"])

contents.py

@@ -27,20 +27,49 @@ how_to_use = r"""
 citation = r"""
 <details>
     <summary><h3 class="summary-label">📚 Citing PECoRe</h3></summary>
+    <p>To refer to the PECoRe framework for context usage detection, cite:</p>
+<div class="code_wrap"><button class="copy_code_button" title="copy">
+<span class="copy-text"><svg viewBox="0 0 32 32" height="100%" width="100%" xmlns="http://www.w3.org/2000/svg"><path d="M28 10v18H10V10h18m0-2H10a2 2 0 0 0-2 2v18a2 2 0 0 0 2 2h18a2 2 0 0 0 2-2V10a2 2 0 0 0-2-2Z" fill="currentColor"></path><path d="M4 18H2V4a2 2 0 0 1 2-2h14v2H4Z" fill="currentColor"></path></svg></span>
+<span class="check"><svg stroke-linejoin="round" stroke-linecap="round" stroke-width="3" stroke="currentColor" fill="none" viewBox="0 0 24 24" height="100%" width="100%" xmlns="http://www.w3.org/2000/svg"><polyline points="20 6 9 17 4 12"></polyline></svg></span>
+</button><pre><code>
+@inproceedings{sarti-etal-2023-quantifying,
+    title = "Quantifying the Plausibility of Context Reliance in Neural Machine Translation",
+    author = "Sarti, Gabriele and
+        Chrupa{\l}a, Grzegorz and
+        Nissim, Malvina and
+        Bisazza, Arianna",
+    booktitle = "The Twelfth International Conference on Learning Representations (ICLR 2024)",
+    month = may,
+    year = "2024",
+    address = "Vienna, Austria",
+    publisher = "OpenReview",
+    url = "https://openreview.net/forum?id=XTHfNGI3zT"
+}
+</code></pre></div>
 
-    @inproceedings{sarti-etal-2023-quantifying,
-        title = "Quantifying the Plausibility of Context Reliance in Neural Machine Translation",
-        author = "Sarti, Gabriele and
-            Chrupa{\l}a, Grzegorz and
-            Nissim, Malvina and
-            Bisazza, Arianna",
-        booktitle = "The Twelfth International Conference on Learning Representations (ICLR 2024)",
-        month = may,
-        year = "2024",
-        address = "Vienna, Austria",
-        publisher = "OpenReview",
-        url = "https://openreview.net/forum?id=XTHfNGI3zT"
-    }
+
+If you use the Inseq implementation of PECoRe (<a href="https://inseq.org/en/latest/main_classes/cli.html#attribute-context"><code>inseq attribute-context</code></a>), please also cite:
+<div class="code_wrap"><button class="copy_code_button" title="copy">
+<span class="copy-text"><svg viewBox="0 0 32 32" height="100%" width="100%" xmlns="http://www.w3.org/2000/svg"><path d="M28 10v18H10V10h18m0-2H10a2 2 0 0 0-2 2v18a2 2 0 0 0 2 2h18a2 2 0 0 0 2-2V10a2 2 0 0 0-2-2Z" fill="currentColor"></path><path d="M4 18H2V4a2 2 0 0 1 2-2h14v2H4Z" fill="currentColor"></path></svg></span>
+<span class="check"><svg stroke-linejoin="round" stroke-linecap="round" stroke-width="3" stroke="currentColor" fill="none" viewBox="0 0 24 24" height="100%" width="100%" xmlns="http://www.w3.org/2000/svg"><polyline points="20 6 9 17 4 12"></polyline></svg></span>
+</button><pre><code>
+@inproceedings{sarti-etal-2023-inseq,
+    title = "Inseq: An Interpretability Toolkit for Sequence Generation Models",
+    author = "Sarti, Gabriele  and
+    Feldhus, Nils  and
+    Sickert, Ludwig  and
+    van der Wal, Oskar and
+    Nissim, Malvina and
+    Bisazza, Arianna",
+    booktitle = "Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics (Volume 3: System Demonstrations)",
+    month = jul,
+    year = "2023",
+    address = "Toronto, Canada",
+    publisher = "Association for Computational Linguistics",
+    url = "https://aclanthology.org/2023.acl-demo.40",
+    pages = "421--435",
+}
+</code></pre></div>
 
 </details>
 """