anuragshas commited on
Commit
524d26c
1 Parent(s): ff2143d

Update eval.py

Browse files
Files changed (1) hide show
  1. eval.py +1 -1
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):