Jan van Doorn commited on
Commit
100b19d
·
unverified ·
1 Parent(s): 6056819

changed to one example

Browse files
Files changed (1) hide show
  1. app.py +7 -1
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 = [ [{'sampling_rate': atco2[i]['audio']['sampling_rate'], 'raw': atco2[i]['audio']['array']}, False, 'large-v3'] for i in range(num_examples)]
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):