Kamalaskar Disha Vinay (EXT) commited on
Commit ·
ea914b0
1
Parent(s): db17692
fixing issue
Browse files
app.py
CHANGED
|
@@ -11,8 +11,7 @@ import tempfile
|
|
| 11 |
import os
|
| 12 |
from huggingface_hub import hf_hub_download
|
| 13 |
import whisper
|
| 14 |
-
|
| 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 |
-
|
| 163 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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)}")
|