Spaces:

rbarman
/

openvino_image_classification

Runtime error

rbarman commited on Oct 4, 2022

Commit

9f7d4e9

1 Parent(s): 768d600

extra paren

Files changed (1) hide show

app.py CHANGED Viewed

@@ -37,7 +37,7 @@ def predict(img: np.ndarray) -> str:
     imagenet_classes = ['background'] + imagenet_classes
     best_class = imagenet_classes[result_index]
     # clean up
-    best_class = best_class.partition(' ')[2])
     # TODO: get n best results with corresponding probabilities?
     return best_class

     imagenet_classes = ['background'] + imagenet_classes
     best_class = imagenet_classes[result_index]
     # clean up
+    best_class = best_class.partition(' ')[2]
     # TODO: get n best results with corresponding probabilities?
     return best_class