Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import os
+import subprocess
 import torch
 import gc
 from diffusers import AutoencoderKLCogVideoX, CogVideoXImageToVideoPipeline, CogVideoXTransformer3DModel
@@ -109,16 +110,20 @@ def infer(image_path, prompt, orbit_type, progress=gr.Progress(track_tqdm=True))
     export_to_video(video.frames[0], temp_path, fps=8)
     
     try:
-        # Use ffmpeg-python
-        stream = ffmpeg.input(temp_path)
-        stream = ffmpeg.filter(stream, 'scale', target_width, target_height)
-        stream = ffmpeg.output(stream, final_path, 
-                             vcodec='libx264', 
-                             preset='medium', 
-                             crf=23)
-        ffmpeg.run(stream, overwrite_output=True)
-    except Exception as e:
-        print(f'Error during video processing: {str(e)}')
+        # Use ffmpeg via subprocess
+        cmd = [
+            'ffmpeg',
+            '-i', temp_path,
+            '-vf', f'scale={target_width}:{target_height}',
+            '-c:v', 'libx264',
+            '-preset', 'medium',
+            '-crf', '23',
+            '-y',  # Overwrite output file if it exists
+            final_path
+        ]
+        subprocess.run(cmd, check=True, capture_output=True)
+    except subprocess.CalledProcessError as e:
+        print(f"FFmpeg error: {e.stderr.decode()}")
         raise e
     finally:
         if os.path.exists(temp_path):