Junfeng5 commited on
Commit
73c8314
1 Parent(s): 2804a9f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -616,7 +616,7 @@ def segment_video(video, prompt_mode, categoryname, custom_category, expressiong
616
  size = (ori_width,ori_height)
617
  print('writing video...')
618
  output_file = "test.mp4"
619
- writer = skvideo.io.FFmpegWriter(output_file,outputdict={'-r': read_fps,})
620
  for i in range(len(ourput_frames)):
621
  writer.writeFrame(ourput_frames[i])
622
  writer.close()
@@ -841,7 +841,7 @@ def segment_video(video, prompt_mode, categoryname, custom_category, expressiong
841
 
842
  size = (ori_width,ori_height)
843
  output_file = "test.mp4"
844
- writer = skvideo.io.FFmpegWriter(output_file,outputdict={'-r': read_fps,})
845
  for i in range(len(output_vos_results)):
846
  writer.writeFrame(output_vos_results[i])
847
  writer.close()
 
616
  size = (ori_width,ori_height)
617
  print('writing video...')
618
  output_file = "test.mp4"
619
+ writer = skvideo.io.FFmpegWriter(output_file,outputdict={'-r': str(read_fps),})
620
  for i in range(len(ourput_frames)):
621
  writer.writeFrame(ourput_frames[i])
622
  writer.close()
 
841
 
842
  size = (ori_width,ori_height)
843
  output_file = "test.mp4"
844
+ writer = skvideo.io.FFmpegWriter(output_file,outputdict={'-r': str(read_fps),})
845
  for i in range(len(output_vos_results)):
846
  writer.writeFrame(output_vos_results[i])
847
  writer.close()