Update pipeline.py
Browse files- pipeline.py +1 -2
pipeline.py
CHANGED
|
@@ -19,8 +19,7 @@ class PreTrainedPipeline():
|
|
| 19 |
def __call__(self, inputs: "Image.Image")-> List[Dict[str, Any]]:
|
| 20 |
|
| 21 |
# convert img to numpy array, resize and normalize to make the prediction
|
| 22 |
-
|
| 23 |
-
img = np.array(inputs)
|
| 24 |
|
| 25 |
im = tf.image.resize(img, (128, 128))
|
| 26 |
im = tf.cast(im, tf.float32) / 255.0
|
|
|
|
| 19 |
def __call__(self, inputs: "Image.Image")-> List[Dict[str, Any]]:
|
| 20 |
|
| 21 |
# convert img to numpy array, resize and normalize to make the prediction
|
| 22 |
+
img = np.array(inputs)
|
|
|
|
| 23 |
|
| 24 |
im = tf.image.resize(img, (128, 128))
|
| 25 |
im = tf.cast(im, tf.float32) / 255.0
|