Update app.py

app.py

@@ -301,7 +301,15 @@ def inference_and_return_video(dilation_iterations, num_inference_steps, video_p
 
     video_file = f"/tmp/{time.time()}-{random.random()}-removed_output.mp4"
     clip = ImageSequenceClip(list(out), fps=fps)
-    clip.write_videofile(video_file, codec="libx264", audio=False, verbose=False, logger=None)
+    clip.write_videofile(
+        video_file,
+        codec="libx264",
+        audio=False,
+        fps=fps,
+        preset="medium",
+        ffmpeg_params=["-crf", "17", "-pix_fmt", "yuv420p"],
+        verbose=False, logger=None
+    )
     return video_file
 
 @spaces.GPU(duration=40)