Update mai.py
Browse files
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})
|