Spaces:
Paused
Paused
Update app.py
Browse files
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:
|