Spaces:

chansung
/

testtest

Runtime error

chansung commited on Jul 24, 2022

Commit

e607f25

1 Parent(s): e97a1d6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -228,7 +228,7 @@ def sepia(input_img):
     color_seg = color_seg[..., ::-1]
     # Show image + mask
-    pred_img = np.array(input_img) * 0.5 + color_seg * 0.5
     pred_img = pred_img.astype(np.uint8)
     fig = draw_plot(pred_img, seg)

     color_seg = color_seg[..., ::-1]
     # Show image + mask
+    pred_img = img * 0.5 + color_seg * 0.5
     pred_img = pred_img.astype(np.uint8)
     fig = draw_plot(pred_img, seg)