Upload compare_gradio.py

compare_gradio.py

@@ -34,7 +34,19 @@ headers = {"X-API-Key": API_KEY, "Content-Type": "application/json"}
 
 
 def update_models():
-    models = random.sample(["pro", "pro2", "turbo", "turbo2"], 2)
+
+    models_pro = ["pro", "pro2", "pro2-fast"]
+    models_turbo = ["turbo", "turbo2", "turbo2-fast"]
+    
+    # Randomly select if we want to use pro or turbo models
+    if random.random() < 0.5:
+        models = models_pro
+    else:
+        models = models_turbo
+
+    # Randomly select two models
+    models = random.sample(models, 2)
+
     print(f"Models updated: {models}")
     return models