YassineB commited on
Commit
6207ffd
1 Parent(s): c3ac0f4

fix output

Browse files
Files changed (1) hide show
  1. 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
- return prediction
 
 
30
 
31
  batch = self.preprocess(img).unsqueeze(0)
32
  prediction = self.pipeline(batch).squeeze(0).softmax(0)
33
+
34
+ return prediction.tolist()