update

pages/arena.py

@@ -11,20 +11,18 @@ from utils.multiple_stream import stream_data
 from pages.summarization_playground import get_model_batch_generation
 
 def create_arena():
+    with open("prompt/prompt.json", "r") as file:
+        json_data = file.read()
+        prompts = json.loads(json_data)
+
     with gr.Blocks() as demo:
         with gr.Group():
-            gr.Markdown("## This is a playground to test prompts for clinical dialogue summarizations")
-
-            with open("prompt/prompt.json", "r") as file:
-                json_data = file.read()
-                prompts = json.loads(json_data)
-
             datapoint = random.choice(dataset)
             datapoint = datapoint['section_text'] + '\n\nDialogue:\n' + datapoint['dialogue']
             submit_button = gr.Button("✨ Submit ✨")
     
             with gr.Row():
-                columns = [gr.Textbox(label=f"Column {i+1}", lines=10) for i in range(3)]
+                columns = [gr.Textbox(label=f"Column {i+1}", lines=10) for i in range(len(prompts))]
             
             content_list = [prompt + '\n{' + datapoint + '}\n\nsummary:' for prompt in prompts]
             model = get_model_batch_generation("Qwen/Qwen2-1.5B-Instruct")