sanchit-gandhi HF staff commited on
Commit
5b3945c
1 Parent(s): 7781396

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -15,6 +15,9 @@ import soundfile as sf
15
 
16
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
17
 
 
 
 
18
  # Intialise STT (Whisper)
19
  pipe = pipeline(
20
  task="automatic-speech-recognition",
 
15
 
16
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
17
 
18
+ print(f"Is CUDA available: {torch.cuda.is_available()}")
19
+ print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
20
+
21
  # Intialise STT (Whisper)
22
  pipe = pipeline(
23
  task="automatic-speech-recognition",