Hugging Face's logo

Spaces:
whitecircle-ai
/
circle-guard-bench
Running

App Files Community
1
apsys commited on
Commit
2204f91
·
1 Parent(s): e478b36

refactor

Browse files
Files changed (1) hide show
  1. src/display/utils.py +2 -2
src/display/utils.py CHANGED
@@ -39,9 +39,9 @@ class ModelType(Enum):
39
  class GuardModelType(str, Enum):
40
  """Guard model types for the leaderboard."""
41
  LLAMA_GUARD = "llama_guard"
42
- PROMPT_GUARD_CLF = "prompt_guard_clf"
43
  ATLA_SELENE = "atla_selene"
44
- GEMMA_SHIELD = "gemma_shield"
45
  LLM_REGEXP = "llm_regexp"
46
  LLM_SO = "llm_so"
47
  WC_GUARD = "wc_guard"
 
39
  class GuardModelType(str, Enum):
40
  """Guard model types for the leaderboard."""
41
  LLAMA_GUARD = "llama_guard"
42
+ CLASSIFIER = "classifier"
43
  ATLA_SELENE = "atla_selene"
44
+ OPENAI_MODERATION = "openai_moderation"
45
  LLM_REGEXP = "llm_regexp"
46
  LLM_SO = "llm_so"
47
  WC_GUARD = "wc_guard"