Update app.py

app.py

@@ -75,13 +75,12 @@ def create_demo():
                 minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"
             )
 
-        # Model Selection Section (limit to two models)
+        # Model Selection Section (no maximum_choices)
         with gr.Row():
             selected_models = gr.CheckboxGroup(
                 choices=list(model_options.keys()),
                 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
@@ -107,6 +106,8 @@ def create_demo():
         def generate_responses(
             message, history, system_message, max_tokens, temperature, top_p, selected_models
         ):
+            if len(selected_models) != 2:
+                return "Error: Please select exactly two models to compare."
             responses = respond(
                 message, history, system_message, max_tokens, temperature, top_p, selected_models
             )