Update tasks/text.py

tasks/text.py

@@ -16,7 +16,7 @@ DESCRIPTION = "GTE Architecture"
 ROUTE = "/text"
 
 class AutoBertClassifier(nn.Module):
-    def __init__(self, num_labels=8, model_path="TaylorAI/gte-tiny"):
+    def __init__(self, num_labels=8, model_path="prajjwal1/bert-tiny"):
         super().__init__()
         self.tokenizer = AutoTokenizer.from_pretrained(model_path)
         self.bert = AutoModel.from_pretrained(model_path)