Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -5,6 +5,7 @@ from transformers import pipeline
|
|
5 |
import tempfile
|
6 |
import os
|
7 |
import uuid
|
|
|
8 |
|
9 |
MODEL_NAME = "ylacombe/whisper-large-v3-turbo"
|
10 |
BATCH_SIZE = 8
|
@@ -24,9 +25,11 @@ def transcribe(inputs, previous_transcription):
|
|
24 |
filename = f"{uuid.uuid4().hex}.wav"
|
25 |
filepath = os.path.join(tempfile.gettempdir(), filename)
|
26 |
|
|
|
|
|
|
|
27 |
# Save the audio data to the temporary file
|
28 |
-
|
29 |
-
f.write(inputs[1])
|
30 |
|
31 |
previous_transcription += pipe(filepath, batch_size=BATCH_SIZE, generate_kwargs={"task": "transcribe"}, return_timestamps=True)["text"]
|
32 |
|
|
|
5 |
import tempfile
|
6 |
import os
|
7 |
import uuid
|
8 |
+
import scipy.io.wavfile
|
9 |
|
10 |
MODEL_NAME = "ylacombe/whisper-large-v3-turbo"
|
11 |
BATCH_SIZE = 8
|
|
|
25 |
filename = f"{uuid.uuid4().hex}.wav"
|
26 |
filepath = os.path.join(tempfile.gettempdir(), filename)
|
27 |
|
28 |
+
# Extract sample rate and audio data from the tuple
|
29 |
+
sample_rate, audio_data = inputs
|
30 |
+
|
31 |
# Save the audio data to the temporary file
|
32 |
+
scipy.io.wavfile.write(filepath, sample_rate, audio_data)
|
|
|
33 |
|
34 |
previous_transcription += pipe(filepath, batch_size=BATCH_SIZE, generate_kwargs={"task": "transcribe"}, return_timestamps=True)["text"]
|
35 |
|