Update app.py

app.py

@@ -10,10 +10,10 @@ login(api_key)
 # Predefined list of models to compare (can be expanded)
 model_options = {
     "Llama-3.1-70B": "meta-llama/Llama-3.1-70B-Instruct",
-    "GPT-4": "TheBloke/Open_Gpt4_8x7B-GGUF",
-    "Falcon-40B": "tiiuae/falcon-40b-instruct",
-    "Mistral-7B": "mistralai/Mistral-7B-Instruct-v0.3",
-    "Bloom": "bigscience/bloom",
+    "Qwen-2.5-1.5B-Instruct": "Qwen/Qwen2.5-1.5B-Instruct",
+    "Llama-3.2-1B": "meta-llama/Llama-3.2-1B",
+    "DeepSeek-V2.5": "deepseek-ai/DeepSeek-V2.5",
+    "Athene-V2-Chat": "Nexusflow/Athene-V2-Chat",
 }
 
 # Initialize clients for models
@@ -77,7 +77,7 @@ def create_demo():
             selected_models = gr.CheckboxGroup(
                 choices=list(model_options.keys()),
                 label="Select models to compare",
-                value=["Llama-3.1-70B", "GPT-4"],  # Default models
+                value=["Llama-3.1-70B", "Qwen-2.5-1.5B-Instruct"],  # Default models
             )
 
         submit_button = gr.Button("Generate Responses")
@@ -126,7 +126,7 @@ def create_demo():
             outputs=[response[2] for response in response_boxes],
         )
 
-        vote_state = gr.State([0] * len(model_options))  # One vote count per model
+        vote_state = gr.State([0] * len(model_options))  # Initialize one vote count per model
         
         for model_name, _, _, vote_button, vote_count in response_boxes:
             vote_button.click(