color change

app.py

@@ -34,12 +34,13 @@ def categorize_model(model):
     else:
         return "Other"
     
+# Define a color map with yellow for "H2OVLs"
+color_map = {"H2OVLs": "#FFE600", "Other": "#0047AB", "traditional ocr models": "#00A36C"}
 
 # Apply the categorization to create a new column
 data_valid["Category"] = data_valid["Model"].apply(categorize_model) 
 
 
-
 # ploting 
 def plot_metric(selected_metric):
     filtered_data = data_valid[data_valid[selected_metric] !=0 ]
@@ -51,7 +52,8 @@ def plot_metric(selected_metric):
         y=selected_metric,
         text="Model",
         color="Category",  # Different color for Special and Other categories
-        title=f"{selected_metric} vs Model Size"
+        title=f"{selected_metric} vs Model Size",
+        color_discrete_map=color_map
     )
     
     fig.update_traces(marker=dict(size=10), mode='markers+text', textposition="middle right", textfont=dict(size=10))