Update app.py

app.py

@@ -44,12 +44,12 @@ def tog_off():
 def pl_fn(cap,cur,last,fps,pl_tog):
     player.set(cv2.CAP_PROP_POS_FRAMES, cur)
     ret, frame_bgr = player.read(cur)
-    #frame = cv2.cvtColor(frame_bgr, cv2.COLOR_BGR2RGB)
-    frame = Image.fromarray(frame_bgr)
+    frame = cv2.cvtColor(frame_bgr, cv2.COLOR_BGR2RGB)
+    frame = Image.fromarray(frame)
     output = segmentor(frame)
     for i in range(len(output)):
         mask = np.array(output[i]['mask'])/255
-        img = np.array(frame_bgr)
+        img = np.array(frame)
         img2 = np.zeros_like(img)
         img2[:,:,0] = mask
         img2[:,:,1] = mask