Update app.py

app.py

@@ -120,15 +120,15 @@ def rem_bg(vid,frame_count):
         print (i)
         cv2.imwrite(f'{uid}-rembg/{i}.png',out)
         fbox2.append(f'{uid}-rembg/{i}.png')
-        
-    #capture.set(cv2.CAP_PROP_POS_FRAMES, int(frame_count)-1)
-    #ret, frame_f = capture.read()
-    #frame_ff = cv2.cvtColor(frame_f, cv2.COLOR_BGR2RGB)
-    #cv2.imwrite(f'{uid}/{int(frame_count)-1}.png',frame_ff)
-    #fbox.append(f'{uid}/{int(frame_count)-1}.png')
-    #capture.release()
-    #out = cv2.VideoWriter('video.avi',cv2.VideoWriter_fourcc(*'DIVX'), 15, size)
-    return fbox2
+        yield fbox2
+        #capture.set(cv2.CAP_PROP_POS_FRAMES, int(frame_count)-1)
+        #ret, frame_f = capture.read()
+        #frame_ff = cv2.cvtColor(frame_f, cv2.COLOR_BGR2RGB)
+        #cv2.imwrite(f'{uid}/{int(frame_count)-1}.png',frame_ff)
+        #fbox.append(f'{uid}/{int(frame_count)-1}.png')
+        #capture.release()
+        #out = cv2.VideoWriter('video.avi',cv2.VideoWriter_fourcc(*'DIVX'), 15, size)
+        #return fbox2