tdurbor commited on
Commit
23a56a2
1 Parent(s): 33d08fb

Revert "change dataset and include clipdrop to leaderboard"

Browse files

This reverts commit afef9c8faddd0d8280de20868f631598aa986ec9.

Files changed (2) hide show
  1. app.py +2 -2
  2. db.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ huggingface_token = os.getenv("HUGGINGFACE_HUB_TOKEN")
30
  logging.basicConfig(level=logging.INFO)
31
 
32
  # Load datasets and initialize database
33
- dataset = load_dataset("bgsys/background-removal-arena-green-v0.1", split='train')
34
  fill_database_once()
35
 
36
  # Directory setup for JSON dataset
@@ -62,7 +62,7 @@ def update_rankings_table():
62
  model_vote_counts = compute_votes_per_model()
63
  try:
64
  # Create a list of models to iterate over
65
- models = ["Clipdrop", "Photoroom", "RemoveBG", "BRIA RMBG 2.0"]
66
  rankings = []
67
 
68
  for model in models:
 
30
  logging.basicConfig(level=logging.INFO)
31
 
32
  # Load datasets and initialize database
33
+ dataset = load_dataset("bgsys/background-removal-arena-green", split='train')
34
  fill_database_once()
35
 
36
  # Directory setup for JSON dataset
 
62
  model_vote_counts = compute_votes_per_model()
63
  try:
64
  # Create a list of models to iterate over
65
+ models = ["Photoroom", "RemoveBG", "BRIA RMBG 2.0"]
66
  rankings = []
67
 
68
  for model in models:
db.py CHANGED
@@ -89,7 +89,7 @@ def get_all_votes():
89
 
90
  # Function to compute Elo scores
91
  def compute_elo_scores():
92
- valid_models = ["Photoroom", "RemoveBG", "BRIA RMBG 2.0", "Clipdrop"]
93
 
94
  with SessionLocal() as db:
95
  votes = db.query(Vote).all()
 
89
 
90
  # Function to compute Elo scores
91
  def compute_elo_scores():
92
+ valid_models = ["Photoroom", "RemoveBG", "BRIA RMBG 2.0"]
93
 
94
  with SessionLocal() as db:
95
  votes = db.query(Vote).all()