Spaces:

hoguseki
/

kosbi-bias-detector-api

Sleeping

hoguseki commited on 22 days ago

Commit

4c89831

verified ·

1 Parent(s): f7f14ab

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -52,4 +52,4 @@ class FramingBiasClassifier(nn.Module):
     def forward(self, input_ids, attention_mask, **kwargs):
         out = self.encoder(input_ids=input_ids, attention_mask=attention_mask)
-        return self.classifier(out.last_hidden_state[:, 0, :])

     def forward(self, input_ids, attention_mask, **kwargs):
         out = self.encoder(input_ids=input_ids, attention_mask=attention_mask)
+        return self.classifier(out.last_hidden_state[:, 0, :])