xinchen9 commited on
Commit
4cf0a64
1 Parent(s): eceb25a

[Update]search table

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -99,7 +99,7 @@ def update_table(
99
  # df = select_columns(filtered_df, columns)
100
  filtered_df = hidden_df.copy()
101
 
102
- filtered_df = filtered_df[filtered_df['Models'].isin(open_query)]
103
  # map_open = {'open': 'Yes', 'closed': 'No'}
104
  # filtered_df = filtered_df[filtered_df['Open?'].isin([map_open[o] for o in open_query])]
105
 
@@ -113,7 +113,7 @@ def update_table(
113
 
114
 
115
  def search_table(df: pd.DataFrame, query: str) -> pd.DataFrame:
116
- return df[(df['Model'].str.contains(query, case=False))]
117
 
118
 
119
  def filter_queries(query: str, filtered_df: pd.DataFrame) -> pd.DataFrame:
 
99
  # df = select_columns(filtered_df, columns)
100
  filtered_df = hidden_df.copy()
101
 
102
+ filtered_df = filtered_df[filtered_df['Unlearned_Methods'].isin(open_query)]
103
  # map_open = {'open': 'Yes', 'closed': 'No'}
104
  # filtered_df = filtered_df[filtered_df['Open?'].isin([map_open[o] for o in open_query])]
105
 
 
113
 
114
 
115
  def search_table(df: pd.DataFrame, query: str) -> pd.DataFrame:
116
+ return df[(df['Unlearned_Methods'].str.contains(query, case=False))]
117
 
118
 
119
  def filter_queries(query: str, filtered_df: pd.DataFrame) -> pd.DataFrame: