elucidator8918 commited on
Commit
efbdf10
·
verified ·
1 Parent(s): 950ba1c

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. 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)