Wataru commited on
Commit
0871d92
1 Parent(s): b684373

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -9,8 +9,9 @@ transcribe = pipeline(
9
  chunk_length_s=30,
10
  device='cuda',
11
  )
12
-
13
  transcribe.model.config.forced_decoder_ids = transcribe.tokenizer.get_decoder_prompt_ids(language="ja", task="transcribe")
 
 
14
  @spaces.GPU
15
  def main(audio_path):
16
  return transcribe(audio_path)["text"]
 
9
  chunk_length_s=30,
10
  device='cuda',
11
  )
 
12
  transcribe.model.config.forced_decoder_ids = transcribe.tokenizer.get_decoder_prompt_ids(language="ja", task="transcribe")
13
+ transcribe.to('cuda')
14
+
15
  @spaces.GPU
16
  def main(audio_path):
17
  return transcribe(audio_path)["text"]