Fix Bug

app.py

@@ -1114,7 +1114,7 @@ def run_model(model_name):
         
         # Plotear las muestras reales, agrupadas por label
         # for label in unique_subsets["real"]:
-        for label in result["classes_k"]:
+        for label_k, label in zip(result["classes_k"], unique_subsets["real"]):
             subset = df_all["real"][df_all["real"]['label'] == label]
             source = ColumnDataSource(data={
                 'x': subset['x'],
@@ -1125,7 +1125,7 @@ def run_model(model_name):
             fig_all.circle('x', 'y', size=10,
                         fill_color=color_maps["real"][label],
                         line_color=color_maps["real"][label],
-                        legend_label=f"Real: {label}",
+                        legend_label=f"Real: {label_k}",
                         source=source)
         
         show_real_only = st.checkbox("Show only real samples", value=True, key=f"show_real_only_{model_name}")