Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -39,7 +39,7 @@ class EndpointHandler():
|
|
39 |
serializable_results = {
|
40 |
'boxes': results['boxes'].tolist(),
|
41 |
'scores': results['scores'].tolist(),
|
42 |
-
'
|
43 |
}
|
44 |
return serializable_results
|
45 |
#prediction = self.pipeline(image=[image], candidate_labels=inputs["candidates"])
|
|
|
39 |
serializable_results = {
|
40 |
'boxes': results['boxes'].tolist(),
|
41 |
'scores': results['scores'].tolist(),
|
42 |
+
'candidates': results['classes'] # Already serializable
|
43 |
}
|
44 |
return serializable_results
|
45 |
#prediction = self.pipeline(image=[image], candidate_labels=inputs["candidates"])
|