Tonic commited on
Commit
8a29aa8
·
unverified ·
1 Parent(s): dc058e1

add model name

Browse files
Files changed (1) hide show
  1. tasks/text.py +1 -1
tasks/text.py CHANGED
@@ -35,7 +35,7 @@ class TextClassifier:
35
  def __init__(self):
36
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
37
  max_retries = 3
38
- model_name = "answerdotai/ModernBERT-base"
39
 
40
  for attempt in range(max_retries):
41
  try:
 
35
  def __init__(self):
36
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
37
  max_retries = 3
38
+ model_name = "Tonic/climate-guard-toxic-agent"
39
 
40
  for attempt in range(max_retries):
41
  try: