Add missing Clipdrop outputs
#2
by
baubin
- opened
app.py
CHANGED
@@ -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:
|
@@ -96,7 +96,7 @@ def select_new_image():
|
|
96 |
input_image = sample['original_image']
|
97 |
|
98 |
segmented_images = [sample.get(key) for key in ['clipdrop_image', 'bria_image', 'photoroom_image', 'removebg_image']]
|
99 |
-
segmented_sources = ['Clipdrop', '
|
100 |
|
101 |
if segmented_images.count(None) > 2:
|
102 |
logging.error("Not enough segmented images found for: %s. Resampling another image.", sample['original_filename'])
|
|
|
62 |
model_vote_counts = compute_votes_per_model()
|
63 |
try:
|
64 |
# Create a list of models to iterate over
|
65 |
+
models = ['Clipdrop', 'RemoveBG', 'Photoroom', 'BRIA RMBG 2.0']
|
66 |
rankings = []
|
67 |
|
68 |
for model in models:
|
|
|
96 |
input_image = sample['original_image']
|
97 |
|
98 |
segmented_images = [sample.get(key) for key in ['clipdrop_image', 'bria_image', 'photoroom_image', 'removebg_image']]
|
99 |
+
segmented_sources = ['Clipdrop', 'RemoveBG', 'Photoroom', 'BRIA RMBG 2.0']
|
100 |
|
101 |
if segmented_images.count(None) > 2:
|
102 |
logging.error("Not enough segmented images found for: %s. Resampling another image.", sample['original_filename'])
|