roychao19477 commited on
Commit
ba56247
·
1 Parent(s): 98553c1

Upload model

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -60,6 +60,7 @@ import os
60
  import tempfile
61
  from ultralytics import YOLO
62
  from moviepy import ImageSequenceClip
 
63
  from scipy.io import wavfile
64
  from avse_code import run_avse
65
 
@@ -193,7 +194,7 @@ def extract_faces(video_file):
193
  clip = ImageSequenceClip(
194
  [cv2.cvtColor(cv2.resize(f, (224, 224)), cv2.COLOR_BGR2RGB) for f in frames],
195
  fps=fps
196
- )
197
  clip.write_videofile(output_path, codec="libx264", audio=False, fps=25)
198
 
199
  # Save audio from original, resampled to 16kHz
 
60
  import tempfile
61
  from ultralytics import YOLO
62
  from moviepy import ImageSequenceClip
63
+ from moviepy.editor import vfx
64
  from scipy.io import wavfile
65
  from avse_code import run_avse
66
 
 
194
  clip = ImageSequenceClip(
195
  [cv2.cvtColor(cv2.resize(f, (224, 224)), cv2.COLOR_BGR2RGB) for f in frames],
196
  fps=fps
197
+ ).fx(vfx.flip_vertical)
198
  clip.write_videofile(output_path, codec="libx264", audio=False, fps=25)
199
 
200
  # Save audio from original, resampled to 16kHz