Update app.py

app.py

@@ -30,7 +30,12 @@ from time import time as ttime
 from module.mel_processing import spectrogram_torch
 from my_utils import load_audio
 
-device = "cpu"
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+is_half = eval(
+    os.environ.get("is_half", "True" if torch.cuda.is_available() else "False")
+)
+
 tokenizer = AutoTokenizer.from_pretrained(bert_path)
 bert_model = AutoModelForMaskedLM.from_pretrained(bert_path)
 if is_half == True: