DamonDemon commited on
Commit
2ade011
·
1 Parent(s): ab501f0
Files changed (1) hide show
  1. app.py +4 -7
app.py CHANGED
@@ -74,16 +74,13 @@ def select_columns(df: pd.DataFrame, columns_1: list, columns_2: list, columns_
74
  return filtered_df
75
 
76
 
77
- def filter_model1(
78
- df: pd.DataFrame, model_query: list
79
- ) -> pd.DataFrame:
80
  # Show all models
81
  if len(model_query) == 0:
82
  return df
 
83
  filtered_df = df
84
-
85
- filtered_df = filtered_df[filtered_df["Model"].isin(
86
- model_query)]
87
  return filtered_df
88
 
89
 
@@ -107,7 +104,7 @@ with demo:
107
  label="Unlearning Methods",
108
  choices=methods,
109
  interactive=True,
110
- elem_id="filter-columns-type",
111
  )
112
 
113
  with gr.Row():
 
74
  return filtered_df
75
 
76
 
77
+ def filter_model1(df: pd.DataFrame, model_query: list) -> pd.DataFrame:
 
 
78
  # Show all models
79
  if len(model_query) == 0:
80
  return df
81
+
82
  filtered_df = df
83
+ filtered_df = filtered_df[filtered_df["Model"].isin(model_query)]
 
 
84
  return filtered_df
85
 
86
 
 
104
  label="Unlearning Methods",
105
  choices=methods,
106
  interactive=True,
107
+ elem_id="column-select",
108
  )
109
 
110
  with gr.Row():