Changed output format
Browse files- 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]
|