Update model_list.py

model_list.py

@@ -80,8 +80,8 @@ class ModelList:
             df = df[~df.github.isna()]
         if has_model:
             df = df[~df.hub.isna() | ~df.other.isna()]
-        df = df[df.type.isin(set(data_types))]
-        df = df[df.type.isin(set(model_types))]
+        df = df[df.data_type.isin(set(data_types))]
+        df = df[df.base_model.isin(set(model_types))]
         return df
 
     @staticmethod