Update app.py

app.py

@@ -143,7 +143,7 @@ def get_frames(video_in, img_type):
     cv2.destroyAllWindows()
     print("broke the video into frames")
     
-    return frames, fps, clip.fps
+    return frames, fps
 
 def get_matte(video_in, subject_to_remove):
     print("Trying to call video matting")
@@ -223,13 +223,18 @@ def infer_auto(project_name, video_in, subject_to_remove):
 
 
     #bmx_trees_folder = os.path.join(output_folder, "bmx-trees")
+
+    # Convert the float fps to an integer
+    needed_fps = int(video_frames[2])
+
         
     command = [
       f"python", 
       f"inference_propainter.py",
       f"--video={video_frames_folder}",
       f"--mask={masks_folder}",
-      f"--output={output_folder}"
+      f"--output={output_folder}",
+      --save_fps=needed_fps
     ]
 
     execute_command(command)
@@ -250,27 +255,7 @@ def infer_auto(project_name, video_in, subject_to_remove):
     for item in results_folder_content:
         print(item)
 
-
-    # Convert the float fps to an integer
-    needed_fps = int(video_frames[2])
-
-    # Load the videos
-    video1 = VideoFileClip(results_folder_content[0])
-    video2 = VideoFileClip(results_folder_content[1])
-
-    # Set the desired frame rate (e.g., 12 fps)
-    new_fps = needed_fps
-
-    # Convert videos to the new frame rate using subclip
-    video1 = video1.subclip(0, video1.duration).set_fps(new_fps)
-    video2 = video2.subclip(0, video2.duration).set_fps(new_fps)
-
-    # Save the converted videos
-    video1.write_videofile(f'masked_{new_fps}fps.mp4', fps=new_fps)
-    video2.write_videofile(f'final_{new_fps}fps.mp4', fps=new_fps)
-
-
-    return "done", f'masked_{new_fps}fps.mp4', f'final_{new_fps}fps.mp4'
+    return "done", results_folder_content[0], results_folder_content[1]
 
 css="""
 #col-container{