usamadv commited on
Commit
2a8ee29
Β·
verified Β·
1 Parent(s): 5946df8

Update mai.py

Browse files
Files changed (1) hide show
  1. mai.py +1 -1
mai.py CHANGED
@@ -10,7 +10,7 @@ pipe = pipeline(
10
  "automatic-speech-recognition", model="openai/whisper-base", device=device
11
  )
12
 
13
- dataset = load_dataset("facebook/voxpopuli", "es", split="validation", streaming=True)
14
  sample = next(iter(dataset))
15
  def translate(audio):
16
  outputs = pipe(audio, generate_kwargs={"task": "translate",'max_new_tokens':255})
 
10
  "automatic-speech-recognition", model="openai/whisper-base", device=device
11
  )
12
 
13
+ dataset = load_dataset("facebook/voxpopuli", "es", split="validation", streaming=True,trust_remote_code=True)
14
  sample = next(iter(dataset))
15
  def translate(audio):
16
  outputs = pipe(audio, generate_kwargs={"task": "translate",'max_new_tokens':255})