Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -229,7 +229,7 @@ def do_prediction(model_name, img):
|
|
229 |
img_border[:, :] = 1
|
230 |
img_border = np.repeat(img_border[:, :, np.newaxis], 3, axis=2)
|
231 |
|
232 |
-
return "No numerical output", visualize_model_output(img_border,
|
233 |
|
234 |
|
235 |
# bitmap output
|
|
|
229 |
img_border[:, :] = 1
|
230 |
img_border = np.repeat(img_border[:, :, np.newaxis], 3, axis=2)
|
231 |
|
232 |
+
return "No numerical output", visualize_model_output(img_border,img_org, model_name)
|
233 |
|
234 |
|
235 |
# bitmap output
|