Revert "change dataset and include clipdrop to leaderboard"
Browse filesThis reverts commit afef9c8faddd0d8280de20868f631598aa986ec9.
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
|
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 = ["
|
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"
|
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()
|