parsi-ai-nlpclass
/

ParsbertHallu

Text Classification

Model card Files Files and versions Community

Pooya-Fallah commited on Mar 3

Commit

af020ce

•

1 Parent(s): 66db36d

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -26,7 +26,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 class ParsbertHallu(nn.Module, PyTorchModelHubMixin):
     def __init__(self):
         super().__init__()
-        self.transformer_model = transformers.AutoModelForSequenceClassification.from_pretrained("Pooya-Fallah/ParsBERT-nli-FarsTail-FarSick",
                                                                            num_labels=3)
         self.head = nn.Sequential(
             nn.Linear(3,1),

 class ParsbertHallu(nn.Module, PyTorchModelHubMixin):
     def __init__(self):
         super().__init__()
+        self.transformer_model = transformers.AutoModelForSequenceClassification.from_pretrained("parsi-ai-nlpclass/ParsBERT-nli-FarsTail-FarSick",
                                                                            num_labels=3)
         self.head = nn.Sequential(
             nn.Linear(3,1),