feat: display 3 labels with scores per model in separate columns

app.py

@@ -5,7 +5,7 @@ import pandas as pd
 from huggingface_hub import login
 
 MIN_ACEPTABLE_SCORE = 0.1
-MAX_N_LABELS = 5
+MAX_N_LABELS = 3
 
 import os
 
@@ -64,13 +64,16 @@ def classify(image, model_name):
     return result
 
 def format_results(results):
-    labels = []
-    scores = []
+    formatted = []
     for r in results[:MAX_N_LABELS]:
         if r['score'] >= MIN_ACEPTABLE_SCORE:
-            labels.append(r['label'])
-            scores.append(f"{r['score']:.2%}")
-    return "<br>".join(labels), "<br>".join(scores)
+            formatted.append({
+                "label": r['label'],
+                "score": f"{r['score']:.2%}"
+            })
+    while len(formatted) < 3:
+        formatted.append({"label": "-", "score": "-"})
+    return formatted
 
 def main():
     st.title("Image Classification - Compare All Models")
@@ -97,19 +100,27 @@ def main():
                 status_text.text(f"Running model {i+1}/{len(MODELS)}: {model_name}")
                 try:
                     classification_result = classify(image_to_classify, model_name)
-                    labels, scores = format_results(classification_result)
+                    formatted = format_results(classification_result)
                     results_data.append({
                         "Model": model_name,
                         "Category": category,
-                        "Top Labels": labels,
-                        "Scores": scores
+                        "Label 1": formatted[0]["label"],
+                        "Score 1": formatted[0]["score"],
+                        "Label 2": formatted[1]["label"],
+                        "Score 2": formatted[1]["score"],
+                        "Label 3": formatted[2]["label"],
+                        "Score 3": formatted[2]["score"],
                     })
                 except Exception as e:
                     results_data.append({
                         "Model": model_name,
                         "Category": category,
-                        "Top Labels": f"Error: {str(e)[:50]}",
-                        "Scores": "-"
+                        "Label 1": f"Error: {str(e)[:50]}",
+                        "Score 1": "-",
+                        "Label 2": "-",
+                        "Score 2": "-",
+                        "Label 3": "-",
+                        "Score 3": "-",
                     })
                 progress_bar.progress((i + 1) / len(MODELS))
             
@@ -132,8 +143,12 @@ def main():
                     column_config={
                         "Model": st.column_config.TextColumn("Model", width="medium"),
                         "Category": st.column_config.TextColumn("Category", width="small"),
-                        "Top Labels": st.column_config.TextColumn("Top Labels", width="large"),
-                        "Scores": st.column_config.TextColumn("Scores", width="medium"),
+                        "Label 1": st.column_config.TextColumn("Label 1", width="medium"),
+                        "Score 1": st.column_config.TextColumn("Score 1", width="small"),
+                        "Label 2": st.column_config.TextColumn("Label 2", width="medium"),
+                        "Score 2": st.column_config.TextColumn("Score 2", width="small"),
+                        "Label 3": st.column_config.TextColumn("Label 3", width="medium"),
+                        "Score 3": st.column_config.TextColumn("Score 3", width="small"),
                     }
                 )