tdurbor commited on
Commit
1f1720f
1 Parent(s): e587a3a

Repush clipdrop change after fixed dataset

Browse files
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", 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 = ["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-v0.1", 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 = ["Clipdrop", "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"]
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", "Clipdrop"]
93
 
94
  with SessionLocal() as db:
95
  votes = db.query(Vote).all()