xinchen9 commited on
Commit
3c7c636
1 Parent(s): a0a1a45

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -116,9 +116,9 @@ def update_table(
116
 
117
 
118
  def search_table(df: pd.DataFrame, query: str) -> pd.DataFrame:
119
- df = df[(df['Unlearned_Methods'].str.contains(query, case=False))]
120
- df = df[(df['Diffusion_Models'].str.contains(query, case=False))]
121
- return df
122
 
123
 
124
  def filter_queries(query: str, filtered_df: pd.DataFrame) -> pd.DataFrame:
 
116
 
117
 
118
  def search_table(df: pd.DataFrame, query: str) -> pd.DataFrame:
119
+ df1 = df[(df['Unlearned_Methods'].str.contains(query, case=False))]
120
+ df2 = df1[(df['Diffusion_Models'].str.contains(query, case=False))]
121
+ return df2
122
 
123
 
124
  def filter_queries(query: str, filtered_df: pd.DataFrame) -> pd.DataFrame: