Update app.py

app.py

@@ -616,7 +616,7 @@ def segment_video(video, prompt_mode, categoryname, custom_category, expressiong
         size = (ori_width,ori_height)    
         print('writing video...')
         output_file = "test.mp4"
-        writer = skvideo.io.FFmpegWriter(output_file)
+        writer = skvideo.io.FFmpegWriter(output_file,outputdict={'-r': read_fps,})
         for i in range(len(ourput_frames)):
             writer.writeFrame(ourput_frames[i])
         writer.close()
@@ -841,10 +841,14 @@ def segment_video(video, prompt_mode, categoryname, custom_category, expressiong
 
         size = (ori_width,ori_height)    
         output_file = "test.mp4"
-        out = cv2.VideoWriter(output_file,cv2.VideoWriter_fourcc(*'avc1'), read_fps, size) 
+        writer = skvideo.io.FFmpegWriter(output_file,outputdict={'-r': read_fps,})
         for i in range(len(output_vos_results)):
-            out.write(output_vos_results[i])
-        out.release()
+            writer.writeFrame(output_vos_results[i])
+        writer.close()
+        # out = cv2.VideoWriter(output_file,cv2.VideoWriter_fourcc(*'avc1'), read_fps, size) 
+        # for i in range(len(output_vos_results)):
+        #     out.write(output_vos_results[i])
+        # out.release()
         torch.cuda.empty_cache()
         return output_file