Update app.py

app.py

@@ -55,8 +55,8 @@ def create_demo():
         gr.Markdown("# AI Model Comparison Tool 🌟")
         gr.Markdown(
             """
-            Compare responses from multiple AI models side-by-side.  
-            Select models, ask a question, and vote for the best response!
+            Compare responses from two AI models side-by-side.  
+            Select two models, ask a question, and compare their responses in real time!
             """
         )
 
@@ -75,12 +75,13 @@ def create_demo():
                 minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"
             )
 
-        # Model Selection Section
+        # Model Selection Section (limit to two models)
         with gr.Row():
             selected_models = gr.CheckboxGroup(
                 choices=list(model_options.keys()),
-                label="Select models to compare",
+                label="Select two models to compare",
                 value=["Llama-3.1-70B", "Qwen-2.5-1.5B-Instruct"],  # Default models
+                maximum_choices=2,  # Limit to two models
             )
 
         # Dynamic Response Section
@@ -89,7 +90,7 @@ def create_demo():
         def generate_response_boxes(models):
             # Clear current components
             response_boxes.clear()
-            # Add new response boxes dynamically
+            # Add new response boxes dynamically for only two models
             for model_name in models:
                 with gr.Column() as response_column:
                     response_box = gr.Textbox(label=f"Response from {model_name}", interactive=False)