Spaces:
Build error
Build error
minor fixes
Browse files
app.py
CHANGED
|
@@ -18,7 +18,6 @@ def predict(image):
|
|
| 18 |
im = tf.image.resize(img, (128, 128))
|
| 19 |
im = tf.cast(im, tf.float32) / 255.0
|
| 20 |
pred_mask = model.predict(im[tf.newaxis, ...])
|
| 21 |
-
pred_mask = tf.image.resize(pred_mask, shape)
|
| 22 |
|
| 23 |
|
| 24 |
# take the best performing class for each pixel
|
|
|
|
| 18 |
im = tf.image.resize(img, (128, 128))
|
| 19 |
im = tf.cast(im, tf.float32) / 255.0
|
| 20 |
pred_mask = model.predict(im[tf.newaxis, ...])
|
|
|
|
| 21 |
|
| 22 |
|
| 23 |
# take the best performing class for each pixel
|