Omnibus commited on
Commit
372e86a
·
1 Parent(s): b44437d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -69,7 +69,7 @@ def im_2_vid(images,fps):
69
  # video.write(img)
70
 
71
 
72
- return (f'{uid}-rembg/bg_removed-{uid}.mp4')
73
  def rem_bg(vid):
74
  new_video_in = str(vid)
75
  capture = cv2.VideoCapture(new_video_in)
@@ -88,8 +88,8 @@ def rem_bg(vid):
88
  fbox2.append(f'{uid}-rembg/{i}.png')
89
  frame_num=f'Working on {i+1} of {frame_count1-1}'
90
  yield fbox2,frame_num,None
91
-
92
- out_vid = im_2_vid(fbox2, fps)
93
  return (fbox2,frame_num,out_vid)
94
 
95
  def rem_bg_og(inp):
 
69
  # video.write(img)
70
 
71
 
72
+ return (movie_clip, f'{uid}-rembg/bg_removed-{uid}.mp4')
73
  def rem_bg(vid):
74
  new_video_in = str(vid)
75
  capture = cv2.VideoCapture(new_video_in)
 
88
  fbox2.append(f'{uid}-rembg/{i}.png')
89
  frame_num=f'Working on {i+1} of {frame_count1-1}'
90
  yield fbox2,frame_num,None
91
+ frame_num=f'Done: {i+1} of {frame_count1-1}'
92
+ out_vid,_file = im_2_vid(fbox2, fps)
93
  return (fbox2,frame_num,out_vid)
94
 
95
  def rem_bg_og(inp):