devingulliver commited on
Commit
967de3c
·
verified ·
1 Parent(s): dcb5bb7

Can't send it to CUDA at the start?

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers.pipelines.audio_utils import ffmpeg_read
5
  import io
6
  import whisper
7
 
8
- model = whisper.load_model("large", device="cuda")
9
  writer = whisper.utils.get_writer("srt", "/dev/null")
10
 
11
  @spaces.GPU
 
5
  import io
6
  import whisper
7
 
8
+ model = whisper.load_model("large")
9
  writer = whisper.utils.get_writer("srt", "/dev/null")
10
 
11
  @spaces.GPU