Kevin676 commited on
Commit
04f2289
·
1 Parent(s): 0ecdad8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -71,6 +71,7 @@ def transcribe(apikey, upload, audio, choice1):
71
  enhance_model = SpectralMaskEnhancement.from_hparams(
72
  source="speechbrain/metricgan-plus-voicebank",
73
  savedir="pretrained_models/metricgan-plus-voicebank",
 
74
  )
75
 
76
  noisy = enhance_model.load_audio(
 
71
  enhance_model = SpectralMaskEnhancement.from_hparams(
72
  source="speechbrain/metricgan-plus-voicebank",
73
  savedir="pretrained_models/metricgan-plus-voicebank",
74
+ run_opts={"device":"cuda"},
75
  )
76
 
77
  noisy = enhance_model.load_audio(