update model type

app.py

@@ -149,6 +149,15 @@ def prep_df(df):
 
     # replace "model" with "Model" and "model_type" with "Model Type" and "average" with "Average"
     df = df.rename(columns={"model": "Model", "model_type": "Model Type", "average": "Average"})
+
+    # if "Model Type" in columns
+    if "Model Type" in df.columns:
+        # get model_types that have generative in them
+        mask = df["Model Type"].str.contains("generative", case=False, na=False)
+        
+        # set these values to "Generative"
+        df.loc[mask, "Model Type"] = "Generative"
+
     return df
 
 # add count column to all dataframes