Update tasks/text.py

tasks/text.py

@@ -34,7 +34,7 @@ class AutoBertClassifier(nn.Module):
 
 device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
     
-model_repo = "elucidator8918/frugal-ai-text"
+model_repo = "elucidator8918/frugal-ai-text-final"
 model = AutoBertClassifier(num_labels=8)
 model.load_state_dict(load_file(hf_hub_download(repo_id=model_repo, filename="model.safetensors")))
 tokenizer = AutoTokenizer.from_pretrained(model_repo)