fix output
Browse files- handler.py +2 -1
handler.py
CHANGED
@@ -30,4 +30,5 @@ class EndpointHandler():
|
|
30 |
|
31 |
batch = self.preprocess(img).unsqueeze(0)
|
32 |
prediction = self.pipeline(batch).squeeze(0).softmax(0)
|
33 |
-
|
|
|
|
30 |
|
31 |
batch = self.preprocess(img).unsqueeze(0)
|
32 |
prediction = self.pipeline(batch).squeeze(0).softmax(0)
|
33 |
+
|
34 |
+
return prediction.tolist()
|