str label fix

This commit is contained in:
aliabd 2020-07-02 11:17:05 -07:00
parent 8cf88d5079
commit 578619a64d

View File

@ -49,7 +49,7 @@ class Label(AbstractOutput):
def postprocess(self, prediction):
if isinstance(prediction, str):
return {"label": str}
return {"label": prediction}
elif isinstance(prediction, dict):
sorted_pred = sorted(
prediction.items(),