dappradar
/

setfit-collections-multilabel

Text Classification

sentence-transformers

Model card Files Files and versions

Mantas commited on Jun 14, 2023

Commit

a00e348

·

1 Parent(s): d386fa3

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -21,6 +21,6 @@ class EndpointHandler:
             inputs = [inputs]
         # run normal prediction
-        scores = self.model.predict(inputs)[0]
         return [{"label": self.id2label[i], "score": score.item()} for i, score in enumerate(scores)]

             inputs = [inputs]
         # run normal prediction
+        scores = self.model.predict_proba(inputs)[0]
         return [{"label": self.id2label[i], "score": score.item()} for i, score in enumerate(scores)]