add our model type

src/display/utils.py

@@ -44,6 +44,7 @@ class GuardModelType(str, Enum):
     GEMMA_SHIELD = "gemma_shield"
     LLM_REGEXP = "llm_regexp"
     LLM_SO = "llm_so"
+    WC_GUARD = "wc_guard"
 
     def __str__(self):
         """String representation of the guard model type."""