polishing

app.py

@@ -15,20 +15,23 @@ models_data = {
     "Qwen/Qwen2.5-14B" : qwen_dict,
 }
 
+models_no = {
+    "microsoft/phi-4" : 172,
+    "Qwen/Qwen2.5-14B" : 729,
+}
+
 starting_index = 0
 starting_model = [model_name for model_name in models_data.keys()][0]
 
 
-description_text = """
+description_template = """
 This Space is inspired by [Luis Hunt's](https://www.linkedin.com/posts/louiswhunt_see-below-for-6882-pages-of-mmlu-and-gsm8k-activity-7281011488692047872-fWCE?utm_source=share&utm_medium=member_desktop) post.
 He highlights how current top performing models from major vendors are contaminated with benchmark data that is supposed to be used to assess their performance.
 This space aims to partially reproduce this work.
 
 I chose to look at the contamination of **Qwen/Qwen2.5-14B** and **microsoft/phi-4** by **GSM8K** dataset.
 
-For **microsoft/phi-4** I found **172** GSM8K examples that had a least a 0.9 text similarity ratio between generated and original.
-For **Qwen/Qwen2.5-14B** I found **729** GSM8K examples that had a least a 0.9 text similarity ratio between generated and original.
-
+For **{model_name}**, I found **{number}** GSM8K examples that had a least a 0.9 text similarity ratio between generated and original.
 """
 
 
@@ -70,14 +73,15 @@ def change_model(selected_model):
             example["original"], 
             highlighted_output, 
             example["similarity_ratio"],
-            example["seed"]
+            example["seed"],
+            description_template.format(model_name=selected_model, number=models_no[selected_model])
         ]
     )
 
 with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column(scale=1):
-            gr.Markdown(description_text)
+            description_text = gr.Markdown(description_template.format(model_name=starting_model, number=models_no[starting_model]))
         with gr.Column(scale=1):
             pass
     with gr.Row():
@@ -123,7 +127,7 @@ with gr.Blocks() as demo:
                     value=models_data[starting_model][starting_index]["seed"],
                 )
 
-    next_btn = gr.Button("Anoter example")
+    next_btn = gr.Button("Another")
     
     next_btn.click(fn=next_example, 
                    inputs=[selected_model],
@@ -131,6 +135,6 @@ with gr.Blocks() as demo:
 
     selected_model.change(fn=change_model,
                           inputs=[selected_model],
-                          outputs=[prompt, original, output, similarity, seed])
+                          outputs=[prompt, original, output, similarity, seed, description_text])
 
 demo.launch()