[Update]Change models to Diffusion_Models

app.py

@@ -133,7 +133,7 @@ def filter_queries(query: list, filtered_df: pd.DataFrame) -> pd.DataFrame:
     return filtered_df
 
 def search_table_model(df: pd.DataFrame, query: str) -> pd.DataFrame:
-    return df[(df['Models'].str.contains(query, case=False))]
+    return df[(df['Diffusion_Models'].str.contains(query, case=False))]
 
 
 def filter_queries_model(query: str, filtered_df: pd.DataFrame) -> pd.DataFrame: