anonymoussubmitter222 commited on
Commit
4b44857
1 Parent(s): beabbe4

removed cuda references

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -342,7 +342,7 @@ class ASR(sb.core.Brain):
342
 
343
 
344
  from speechbrain.pretrained import EncoderASR,EncoderDecoderASR
345
- french_asr_model = EncoderASR.from_hparams(source="asr-wav2vec2-commonvoice-fr", savedir="pretrained_models/asr-wav2vec2-commonvoice-fr").cuda()
346
  french_asr_model.to("cpu")
347
  cvhparams_file, cvrun_opts, cvoverrides = sb.parse_arguments(["EnglishCV/train_en_with_wav2vec.yaml"])
348
  with open(cvhparams_file) as cvfin:
@@ -410,7 +410,7 @@ def load_paths(wavs_path):
410
 
411
 
412
 
413
- device = 'cuda'
414
  verbose = 0
415
  #FLOW LEVEL FUNCTIONS
416
  def merge_strategy(embeddings1, embeddings2, embeddings3,post1, post2,post3):
 
342
 
343
 
344
  from speechbrain.pretrained import EncoderASR,EncoderDecoderASR
345
+ french_asr_model = EncoderASR.from_hparams(source="asr-wav2vec2-commonvoice-fr", savedir="pretrained_models/asr-wav2vec2-commonvoice-fr")
346
  french_asr_model.to("cpu")
347
  cvhparams_file, cvrun_opts, cvoverrides = sb.parse_arguments(["EnglishCV/train_en_with_wav2vec.yaml"])
348
  with open(cvhparams_file) as cvfin:
 
410
 
411
 
412
 
413
+ device = 'cpu'
414
  verbose = 0
415
  #FLOW LEVEL FUNCTIONS
416
  def merge_strategy(embeddings1, embeddings2, embeddings3,post1, post2,post3):