eb1 commited on
Commit
206fdb3
1 Parent(s): 159d2d1

Update app.py

Browse files

This change is needed in order to successfully run on a GPU

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ def transcribe(audio_file):
9
 
10
 
11
  if torch.cuda.is_available():
12
- device = "gpu"
13
  else:
14
  device = "cpu"
15
  batch_size = 16 # reduce if low on GPU mem
 
9
 
10
 
11
  if torch.cuda.is_available():
12
+ device = "cuda"
13
  else:
14
  device = "cpu"
15
  batch_size = 16 # reduce if low on GPU mem