anuragshas
commited on
Commit
·
524d26c
1
Parent(s):
ff2143d
Update eval.py
Browse files
eval.py
CHANGED
@@ -83,7 +83,7 @@ def main(args):
|
|
83 |
dataset = dataset.cast_column("audio", Audio(sampling_rate=sampling_rate))
|
84 |
|
85 |
# load eval pipeline
|
86 |
-
asr = pipeline("automatic-speech-recognition", model=args.model_id)
|
87 |
|
88 |
# map function to decode audio
|
89 |
def map_to_pred(batch):
|
|
|
83 |
dataset = dataset.cast_column("audio", Audio(sampling_rate=sampling_rate))
|
84 |
|
85 |
# load eval pipeline
|
86 |
+
asr = pipeline("automatic-speech-recognition", model=args.model_id, device=0)
|
87 |
|
88 |
# map function to decode audio
|
89 |
def map_to_pred(batch):
|