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