Spaces:
Running
Running
Jan van Doorn
commited on
changed to one example
Browse files
app.py
CHANGED
@@ -1,16 +1,22 @@
|
|
1 |
from transformers import pipeline
|
2 |
from datasets import load_dataset
|
3 |
import gradio as gr
|
|
|
4 |
import os
|
5 |
|
6 |
atco2 = load_dataset('jlvdoorn/atco2-asr', split='validation')
|
7 |
atcosim = load_dataset('jlvdoorn/atcosim', split='validation')
|
8 |
|
9 |
num_examples = 3
|
10 |
-
examples_atco2 = [
|
11 |
#examples_atcosim = [ [{'sampling_rate': atcosim[i]['audio']['sampling_rate'], 'raw': atcosim[i]['audio']['array']}, False, 'large-v3'] for i in range(num_examples)]
|
12 |
examples = examples_atco2 #+ examples_atcosim
|
13 |
|
|
|
|
|
|
|
|
|
|
|
14 |
whisper = pipeline(model='jlvdoorn/whisper-large-v3-atco2-asr-atcosim')
|
15 |
|
16 |
def transcribe(audio):
|
|
|
1 |
from transformers import pipeline
|
2 |
from datasets import load_dataset
|
3 |
import gradio as gr
|
4 |
+
import numpy as np
|
5 |
import os
|
6 |
|
7 |
atco2 = load_dataset('jlvdoorn/atco2-asr', split='validation')
|
8 |
atcosim = load_dataset('jlvdoorn/atcosim', split='validation')
|
9 |
|
10 |
num_examples = 3
|
11 |
+
examples_atco2 = [[atco2[i]['audio']['array'].astype(np.float32)] for i in range(num_examples)]
|
12 |
#examples_atcosim = [ [{'sampling_rate': atcosim[i]['audio']['sampling_rate'], 'raw': atcosim[i]['audio']['array']}, False, 'large-v3'] for i in range(num_examples)]
|
13 |
examples = examples_atco2 #+ examples_atcosim
|
14 |
|
15 |
+
sr, y = atco2[0]['audio']['sampling_rate'], atco2[0]['audio']['array']
|
16 |
+
y = y.astype(np.float32)
|
17 |
+
y /= np.max(np.abs(y))
|
18 |
+
examples = [[y]]
|
19 |
+
|
20 |
whisper = pipeline(model='jlvdoorn/whisper-large-v3-atco2-asr-atcosim')
|
21 |
|
22 |
def transcribe(audio):
|