Update app.py

app.py

@@ -116,7 +116,9 @@ def update_table(
 
 
 def search_table(df: pd.DataFrame, query: str) -> pd.DataFrame:
-    return df[(df['Unlearned_Methods'].str.contains(query, case=False))]
+    df = df[(df['Unlearned_Methods'].str.contains(query, case=False))]
+    df = df[(df['Diffusion_Models'].str.contains(query, case=False))]
+    return df
 
 
 def filter_queries(query: str, filtered_df: pd.DataFrame) -> pd.DataFrame: