Spaces:

FFZG-cleopatra
/

latvian-twitter-sentiment-classifier

Build error

thak123 commited on May 8, 2023

Commit

55edb3b

1 Parent(s): 20d5db8

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -17,7 +17,7 @@ class BERTBaseUncased(nn.Module):
         print("Model")
     def forward(self, ids, mask, token_type_ids):
-        _, o2 = self.bert(
             ids,
             attention_mask=mask,
             token_type_ids=token_type_ids

         print("Model")
     def forward(self, ids, mask, token_type_ids):
+        output = self.bert(
             ids,
             attention_mask=mask,
             token_type_ids=token_type_ids