Future-Tense commited on
Commit
a765444
1 Parent(s): aaed971

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ def pl_fn(cap,cur,last,fps,pl_tog):
49
  output = segmentor(frame)
50
  for i in range(len(output)):
51
  mask = np.array(output[i]['mask'])/255
52
- img = np.array(image)
53
  img2 = np.zeros_like(img)
54
  img2[:,:,0] = mask
55
  img2[:,:,1] = mask
 
49
  output = segmentor(frame)
50
  for i in range(len(output)):
51
  mask = np.array(output[i]['mask'])/255
52
+ img = np.array(frame_bgr)
53
  img2 = np.zeros_like(img)
54
  img2[:,:,0] = mask
55
  img2[:,:,1] = mask