k-mktr commited on
Commit
efcee16
·
verified ·
1 Parent(s): 2547b76

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -1
app.py CHANGED
@@ -61,7 +61,22 @@ start_backup_thread()
61
 
62
  # Function to get available models (using predefined list)
63
  def get_available_models():
64
- return [model[0] for model in config.get_approved_models()]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
65
 
66
  # Function to get recent opponents for a model
67
  recent_opponents = {}
 
61
 
62
  # Function to get available models (using predefined list)
63
  def get_available_models():
64
+ leaderboard = get_current_leaderboard()
65
+ approved_models = [model[0] for model in config.get_approved_models()]
66
+
67
+ # Filter out models that have reached the battle limit
68
+ active_models = []
69
+ for model in approved_models:
70
+ data = leaderboard.get(model, {})
71
+ total_battles = data.get('wins', 0) + data.get('losses', 0)
72
+ if total_battles < config.MAX_BATTLES_LIMIT:
73
+ active_models.append(model)
74
+
75
+ # If all models are over the limit, fallback to all approved models to avoid errors
76
+ if not active_models:
77
+ return approved_models
78
+
79
+ return active_models
80
 
81
  # Function to get recent opponents for a model
82
  recent_opponents = {}