shhossain
/

all-MiniLM-L6-v2-sentiment-classifier

Text Classification

SententenceTransformerSentimentClassifier

Model card Files Files and versions Community

shhossain commited on Feb 14

Commit

267bbc9

•

1 Parent(s): eb7ffba

Update sentpipeline.py

Files changed (1) hide show

sentpipeline.py +1 -1

sentpipeline.py CHANGED Viewed

@@ -14,7 +14,7 @@ class SentimentModelPipe(Pipeline):
         return self.smodel.encode(inputs, convert_to_tensor=True)
     def postprocess(self, outputs):
-        return outputs.argmax(1).item()
     def _forward(self, tensor):
         with torch.no_grad():

         return self.smodel.encode(inputs, convert_to_tensor=True)
     def postprocess(self, outputs):
+        return outputs
     def _forward(self, tensor):
         with torch.no_grad():