diff --git a/web/src/helpers/render.js b/web/src/helpers/render.js
index 98e7846c..b48ede84 100644
--- a/web/src/helpers/render.js
+++ b/web/src/helpers/render.js
@@ -128,7 +128,10 @@ export const getModelCategories = (() => {
           model.model_name.toLowerCase().includes('babbage') ||
           model.model_name.toLowerCase().includes('davinci') ||
           model.model_name.toLowerCase().includes('curie') ||
-          model.model_name.toLowerCase().includes('ada')
+          model.model_name.toLowerCase().includes('ada') ||
+          model.model_name.toLowerCase().includes('o1') ||
+          model.model_name.toLowerCase().includes('o3') ||
+          model.model_name.toLowerCase().includes('o4')
       },
       anthropic: {
         label: 'Anthropic',