randomtable
commited on
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/*')][:]
|