roychao19477 commited on
Commit
33e4d7d
·
1 Parent(s): f23d724

Update module

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -165,7 +165,11 @@ def extract_faces(video_file):
165
  tmpdir = tempfile.mkdtemp()
166
  output_path = os.path.join(tmpdir, "face_only_video.mp4")
167
  #clip = ImageSequenceClip([cv2.cvtColor(f, cv2.COLOR_BGR2RGB) for f in frames], fps=25)
168
- clip = ImageSequenceClip([cv2.cvtColor(f, cv2.COLOR_BGR2RGB) for f in frames], fps=fps)
 
 
 
 
169
  clip.write_videofile(output_path, codec="libx264", audio=False, fps=25)
170
 
171
  # Save audio from original, resampled to 16kHz
 
165
  tmpdir = tempfile.mkdtemp()
166
  output_path = os.path.join(tmpdir, "face_only_video.mp4")
167
  #clip = ImageSequenceClip([cv2.cvtColor(f, cv2.COLOR_BGR2RGB) for f in frames], fps=25)
168
+ #clip = ImageSequenceClip([cv2.cvtColor(f, cv2.COLOR_BGR2RGB) for f in frames], fps=fps)
169
+ clip = ImageSequenceClip(
170
+ [cv2.cvtColor(cv2.resize(f, (224, 224)), cv2.COLOR_BGR2RGB) for f in frames],
171
+ fps=fps
172
+ )
173
  clip.write_videofile(output_path, codec="libx264", audio=False, fps=25)
174
 
175
  # Save audio from original, resampled to 16kHz