Update app.py
Browse files
app.py
CHANGED
|
@@ -93,7 +93,7 @@ def predict(
|
|
| 93 |
pred = np.repeat(np.expand_dims(pred, axis=-1), 3, axis=-1)
|
| 94 |
image_pred = (pred * np.array(image_pil)).astype(np.uint8)
|
| 95 |
|
| 96 |
-
return
|
| 97 |
|
| 98 |
|
| 99 |
examples = [[_] for _ in glob('examples/*')][:]
|
|
|
|
| 93 |
pred = np.repeat(np.expand_dims(pred, axis=-1), 3, axis=-1)
|
| 94 |
image_pred = (pred * np.array(image_pil)).astype(np.uint8)
|
| 95 |
|
| 96 |
+
return image_pred
|
| 97 |
|
| 98 |
|
| 99 |
examples = [[_] for _ in glob('examples/*')][:]
|