Kamalaskar Disha Vinay (EXT) commited on
Commit
ea914b0
·
1 Parent(s): db17692

fixing issue

Browse files
Files changed (1) hide show
  1. app.py +14 -6
app.py CHANGED
@@ -11,8 +11,7 @@ import tempfile
11
  import os
12
  from huggingface_hub import hf_hub_download
13
  import whisper
14
- from moviepy.editor import VideoFileClip
15
-
16
 
17
  # Configuration
18
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -155,12 +154,21 @@ def extract_video_frames(video_path, max_frames=8, resize=(224, 224)):
155
  return frames
156
 
157
  def extract_audio_from_video(video_path):
158
- """Extract audio from video file"""
159
  try:
160
- video = VideoFileClip(video_path)
161
  audio_path = tempfile.NamedTemporaryFile(delete=False, suffix=".wav").name
162
- video.audio.write_audiofile(audio_path, fps=SAMPLE_RATE, verbose=False, logger=None)
163
- video.close()
 
 
 
 
 
 
 
 
 
 
164
  return audio_path
165
  except Exception as e:
166
  raise ValueError(f"Could not extract audio from video: {str(e)}")
 
11
  import os
12
  from huggingface_hub import hf_hub_download
13
  import whisper
14
+ import subprocess
 
15
 
16
  # Configuration
17
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
154
  return frames
155
 
156
  def extract_audio_from_video(video_path):
157
+ """Extract audio from video file using ffmpeg"""
158
  try:
 
159
  audio_path = tempfile.NamedTemporaryFile(delete=False, suffix=".wav").name
160
+ # Use ffmpeg to extract audio
161
+ import subprocess
162
+ command = [
163
+ 'ffmpeg', '-i', video_path,
164
+ '-vn', # No video
165
+ '-acodec', 'pcm_s16le', # Audio codec
166
+ '-ar', str(SAMPLE_RATE), # Sample rate
167
+ '-ac', '1', # Mono
168
+ '-y', # Overwrite
169
+ audio_path
170
+ ]
171
+ subprocess.run(command, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL, check=True)
172
  return audio_path
173
  except Exception as e:
174
  raise ValueError(f"Could not extract audio from video: {str(e)}")