Update app.py

app.py

@@ -88,7 +88,7 @@ def get_frames(video_in, step, name, resize_w):
 def create_video(frames, fps, type):
     print("building video result")
     clip = ImageSequenceClip(frames, fps=fps)
-    clip.write_videofile(type + "_result.mp4", fps=fps)
+    clip.write_videofile(type + "_result.mp4", fps=fps, codec='libx265', ffmpeg_params=['-c:v', 'libx265'], audio = False)
     
     return type + "_result.mp4"
 
@@ -165,9 +165,15 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
       alpha = 1.0/len(result_frames)
       beta = (1.0 - alpha)
       res = cv2.addWeighted(rgb, alpha, res, beta, 0.0, res)
-      rgb = cv2.cvtColor(rgb, cv2.COLOR_GRAY2RGB)
-      cv2.imwrite(f"opticalfb{i}.jpg", rgb)
-      depth_frames.append(f"opticalfb{i}.jpg")
+      rgb = cv2.cvtColor(rgb, cv2.COLOR_GRAY2RGBA)
+        
+      lo = np.array([0,0,0,0])
+      hi = np.array([0,0,0,255])
+      mask = cv2.inRange(rgb, lo, hi)
+      rgb[mask>0] = (0,0,0,0)
+        
+      cv2.imwrite(f"opticalfb{i}.png", rgb)
+      depth_frames.append(f"opticalfb{i}.png")
       i+=1
       prvs = nxt
 
@@ -222,7 +228,7 @@ with gr.Blocks() as demo:
                 video_input = gr.Video()
                 video_input.change(fn=loadurl, inputs=[video_input], outputs=[url_input])
                 resize_num = gr.Slider(minimum=1, maximum=4096, step=1, value=256, label="Resize to width: ")
-                of_check = gr.Checkbox(value=True, label="Detect motion for depth map: ")
+                of_check = gr.Checkbox(value=True, label="Detect motion to remove background: ")
                 winsize_num = gr.Slider(minimum=1, maximum=256, step=1, value=15, label="Motion detection window size: ")
                 with gr.Row():
                     interpolation_slider = gr.Slider(minimum=1, maximum=5, step=1, value=1, label="Interpolation Steps: ")