Spaces:
Sleeping
Sleeping
Update tasks/text.py
Browse files- tasks/text.py +1 -1
tasks/text.py
CHANGED
@@ -34,7 +34,7 @@ class AutoBertClassifier(nn.Module):
|
|
34 |
|
35 |
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
|
36 |
|
37 |
-
model_repo = "elucidator8918/frugal-ai-text"
|
38 |
model = AutoBertClassifier(num_labels=8)
|
39 |
model.load_state_dict(load_file(hf_hub_download(repo_id=model_repo, filename="model.safetensors")))
|
40 |
tokenizer = AutoTokenizer.from_pretrained(model_repo)
|
|
|
34 |
|
35 |
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
|
36 |
|
37 |
+
model_repo = "elucidator8918/frugal-ai-text-final"
|
38 |
model = AutoBertClassifier(num_labels=8)
|
39 |
model.load_state_dict(load_file(hf_hub_download(repo_id=model_repo, filename="model.safetensors")))
|
40 |
tokenizer = AutoTokenizer.from_pretrained(model_repo)
|