Spaces:
Sleeping
Sleeping
add model name
Browse files- 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 = "
|
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:
|