Update sentpipeline.py
Browse files- sentpipeline.py +2 -7
sentpipeline.py
CHANGED
@@ -21,14 +21,9 @@ class SentimentModelPipe(Pipeline):
|
|
21 |
return self.smodel.encode(inputs, convert_to_tensor=True)
|
22 |
|
23 |
def postprocess(self, outputs):
|
24 |
-
if isinstance(outputs, torch.Tensor):
|
25 |
-
outputs = [outputs]
|
26 |
results = []
|
27 |
-
for
|
28 |
-
|
29 |
-
for i, l in enumerate(out):
|
30 |
-
r.append({"label": self.class_map[i], "score": l.item()})
|
31 |
-
results.append(r)
|
32 |
return results
|
33 |
|
34 |
def _forward(self, tensor):
|
|
|
21 |
return self.smodel.encode(inputs, convert_to_tensor=True)
|
22 |
|
23 |
def postprocess(self, outputs):
|
|
|
|
|
24 |
results = []
|
25 |
+
for i, l in enumerate(outputs):
|
26 |
+
results.append({"label": self.class_map[i], "score": l.item()})
|
|
|
|
|
|
|
27 |
return results
|
28 |
|
29 |
def _forward(self, tensor):
|