Add cuda device check

app.py

@@ -10,6 +10,7 @@ import gradio as gr
 import librosa
 import numpy as np
 import soundfile
+import torch
 
 from edgetts.tts_voices import SUPPORTED_LANGUAGES
 from inference.infer_tool import Svc
@@ -163,6 +164,10 @@ if __name__ == "__main__":
     parser.add_argument("-t", "--temp", default="./workspace")
     args = parser.parse_args()
 
+    print(f"Is CUDA available: {torch.cuda.is_available()}")
+    if torch.cuda.is_available():
+        print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+
     shutil.rmtree(args.temp, ignore_errors=True)
     os.makedirs(args.temp, exist_ok=True)
     TEMPDIR = args.temp