time4et commited on
Commit
9b0689e
·
verified ·
1 Parent(s): 9fbc996

Update audio_processing.py

Browse files
Files changed (1) hide show
  1. audio_processing.py +20 -18
audio_processing.py CHANGED
@@ -1,18 +1,20 @@
1
- import whisper
2
- import tempfile
3
- from pydub import AudioSegment
4
- import os
5
-
6
- # Set FFmpeg path manually (if needed)
7
- AudioSegment.converter = r"C:\ffmpeg\bin\ffmpeg.exe"
8
-
9
- def transcribe_audio(audio_file):
10
- model = whisper.load_model("base")
11
-
12
- # Convert to WAV if necessary
13
- with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as temp_wav:
14
- audio = AudioSegment.from_file(audio_file)
15
- audio.export(temp_wav.name, format="wav")
16
-
17
- result = model.transcribe(temp_wav.name)
18
- return result["text"]
 
 
 
1
+ import whisper
2
+ import tempfile
3
+ from pydub import AudioSegment
4
+ import os
5
+
6
+ # Set FFmpeg path manually (if needed)
7
+ # AudioSegment.converter = r"C:\ffmpeg\bin\ffmpeg.exe"
8
+ AudioSegment.converter = "/usr/bin/ffmpeg"
9
+
10
+
11
+ def transcribe_audio(audio_file):
12
+ model = whisper.load_model("base")
13
+
14
+ # Convert to WAV if necessary
15
+ with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as temp_wav:
16
+ audio = AudioSegment.from_file(audio_file)
17
+ audio.export(temp_wav.name, format="wav")
18
+
19
+ result = model.transcribe(temp_wav.name)
20
+ return result["text"]