davidaf3 commited on
Commit
efc8e18
1 Parent(s): 382cc6e

Changed output format

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -28,4 +28,4 @@ class PreTrainedPipeline():
28
  image = tf.keras.applications.inception_v3.preprocess_input(image)
29
  image = tf.keras.layers.CenterCrop(*self.crop_size)(image)
30
  prediction = self.model(image[tf.newaxis, :])
31
- return {name: float(prediction[name].numpy()[0, 0]) for name in self.nutr_names}
 
28
  image = tf.keras.applications.inception_v3.preprocess_input(image)
29
  image = tf.keras.layers.CenterCrop(*self.crop_size)(image)
30
  prediction = self.model(image[tf.newaxis, :])
31
+ return [{"label": name, "score": float(prediction[name].numpy()[0, 0])} for name in self.nutr_names]