marco/add_mention_birefnet_with_refactor

app.py

@@ -49,6 +49,15 @@ commit_scheduler = CommitScheduler(
     token=huggingface_token
 ) if is_running_in_space() else None
 
+# Model configuration mapping
+MODEL_CONFIG = {
+    'Clipdrop': 'clipdrop_image',
+    'BRIA RMBG 2.0': 'bria_image',
+    'Photoroom': 'photoroom_image',
+    'RemoveBG': 'removebg_image',
+    'BiRefNet v2': 'birefnet_image'
+}
+
 def fetch_elo_scores():
     """Fetch and log Elo scores."""
     try:
@@ -105,14 +114,8 @@ def select_new_image(last_used_indices):
         sample = dataset[random_index]
         input_image = sample['original_image']
 
-        segmented_images = [sample.get(key) for key in [
-            'clipdrop_image', 'bria_image', 'photoroom_image', 
-            'removebg_image', 'birefnet_image'
-        ]]
-        segmented_sources = [
-            'Clipdrop', 'BRIA RMBG 2.0', 'Photoroom', 
-            'RemoveBG', 'BiRefNet v2'
-        ]
+        segmented_images = [sample.get(key) for key in MODEL_CONFIG.values()]
+        segmented_sources = [MODEL_CONFIG.get(key) for key in MODEL_CONFIG.keys()]
         
         if segmented_images.count(None) > 2:
             logging.error("Not enough segmented images found for: %s. Resampling another image.", sample['original_filename'])
@@ -140,10 +143,11 @@ def select_new_image(last_used_indices):
 def get_notice_markdown():
     """Generate the notice markdown with dynamic vote count."""
     total_votes = len(get_all_votes())
+    model_list = ", ".join(MODEL_CONFIG.keys())    
     return f"""
 
     ## 📜 How It Works
-    - **Blind Test**: You will see two images with their background removed from two anonymous background removal models (Clipdrop, RemoveBG, Photoroom, BRIA RMBG 2.0).
+    - **Blind Test**: You will see two images with their background removed from two anonymous background removal models ({model_list}).
     - **Vote for the Best**: Choose the best result, if none stand out choose "Tie". 
 
     ## 📊 Stats