fffiloni commited on
Commit
db305d2
·
1 Parent(s): d41bd3b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import gradio as gr
2
  import subprocess
3
  import imageio
 
4
 
5
  def convert_video(input_file, output_file, format='mp4', codec='libx264'):
6
  try:
@@ -11,8 +12,11 @@ def convert_video(input_file, output_file, format='mp4', codec='libx264'):
11
  # Read video file
12
  reader = imageio.get_reader(input_path)
13
 
 
 
 
14
  # Get writer for output video file with specified codec
15
- writer = imageio.get_writer(output_path, format=format, ffmpeg_params=['-vcodec', codec])
16
 
17
  # Iterate through video frames and write to output file
18
  for frame in reader:
 
1
  import gradio as gr
2
  import subprocess
3
  import imageio
4
+ import imageio_ffmpeg
5
 
6
  def convert_video(input_file, output_file, format='mp4', codec='libx264'):
7
  try:
 
12
  # Read video file
13
  reader = imageio.get_reader(input_path)
14
 
15
+ # Configure FFmpeg parameters
16
+ ffmpeg_params = ['-vcodec', codec]
17
+
18
  # Get writer for output video file with specified codec
19
+ writer = imageio.get_writer(output_path, format=format, ffmpeg_params=ffmpeg_params, plugin='imageio_ffmpeg')
20
 
21
  # Iterate through video frames and write to output file
22
  for frame in reader: