wenxiang guo commited on
Commit
3af9bd5
·
verified ·
1 Parent(s): 5c80cc5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  import spaces
2
  import argparse, os, sys, glob
3
  import pathlib
@@ -15,7 +18,7 @@ from tqdm import tqdm
15
  import preprocess.n2s_by_openai as n2s
16
  from vocoder.bigvgan.models import VocoderBigVGAN
17
  import soundfile
18
- import torchaudio, math
19
  import gradio
20
  import gradio as gr
21
 
 
1
+ import torch
2
+ print(f"Is CUDA available: {torch.cuda.is_available()}")
3
+ print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
4
  import spaces
5
  import argparse, os, sys, glob
6
  import pathlib
 
18
  import preprocess.n2s_by_openai as n2s
19
  from vocoder.bigvgan.models import VocoderBigVGAN
20
  import soundfile
21
+ import math
22
  import gradio
23
  import gradio as gr
24