Add missing Clipdrop outputs

#2
by baubin - opened
Files changed (1) hide show
  1. app.py +2 -2
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 = ["Photoroom", "RemoveBG", "BRIA RMBG 2.0"]
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', 'BRIA RMBG 2.0', 'Photoroom', 'RemoveBG']
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'])