diff --git a/omnivore.py b/omnivore.py
index bfd5eac..6aa49ec 100644
--- a/omnivore.py
+++ b/omnivore.py
@@ -105,5 +105,5 @@ class Omnivore(NNOperator):
         pred_scores, pred_classes = preds.topk(k=self.topk)
         labels = [self.classmap[int(i)] for i in pred_classes[0]]
         scores = [round(float(x), 5) for x in pred_scores[0]]
-
-        return labels, scores, features
+        print(labels,scores)
+        return labels, scores