Pijush2023 commited on
Commit
9cad0ea
·
verified ·
1 Parent(s): 939c5b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -915,6 +915,8 @@ import librosa
915
  from pathlib import Path
916
  import torchaudio
917
 
 
 
918
  # Check if the token is already set in the environment variables
919
  hf_token = os.getenv("HF_TOKEN")
920
  if hf_token is None:
 
915
  from pathlib import Path
916
  import torchaudio
917
 
918
+ device = "cuda:0" if torch.cuda.is_available() else "cpu"
919
+
920
  # Check if the token is already set in the environment variables
921
  hf_token = os.getenv("HF_TOKEN")
922
  if hf_token is None: