THP2903 commited on
Commit
d997553
·
verified ·
1 Parent(s): 836cfdf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -9
app.py CHANGED
@@ -3,22 +3,21 @@ import torch as pt
3
  import torchaudio
4
  import cv2
5
  import os
6
- import subprocess
7
  import numpy as np
8
  import tensorflow as tf
9
  from tensorflow.keras.models import load_model
 
10
 
11
- def convert_video_to_audio_ffmpeg(video_file, output_ext="wav"):
12
- """Converts video to audio directly using `ffmpeg` command with the help of subprocess module"""
13
  filename, ext = os.path.splitext(video_file)
14
- audio_file = f"{filename}.{output_ext}"
15
- subprocess.call(["ffmpeg", "-y", "-i", video_file, audio_file],
16
- stdout=subprocess.DEVNULL,
17
- stderr=subprocess.STDOUT)
18
- return audio_file
19
 
20
  def process_video_audio(video_path):
21
- audio_path = convert_video_to_audio_ffmpeg(video_path)
22
 
23
  wav, sr = torchaudio.load(audio_path)
24
 
 
3
  import torchaudio
4
  import cv2
5
  import os
 
6
  import numpy as np
7
  import tensorflow as tf
8
  from tensorflow.keras.models import load_model
9
+ from moviepy.editor import VideoFileClip
10
 
11
+ def convert_video_to_audio_moviepy(video_file, output_ext="wav"):
12
+ """Converts video to audio using MoviePy library that uses `ffmpeg` under the hood"""
13
  filename, ext = os.path.splitext(video_file)
14
+ clip = VideoFileClip(video_file)
15
+ audio_path = f"{filename}.{output_ext}"
16
+ clip.audio.write_audiofile(audio_path)
17
+ return audio_path
 
18
 
19
  def process_video_audio(video_path):
20
+ audio_path = convert_video_to_audio_moviepy(video_path)
21
 
22
  wav, sr = torchaudio.load(audio_path)
23