Update app.py

app.py

@@ -9,6 +9,8 @@ from num2tib.core import convert
 from num2tib.core import convert2text
 import re
 
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
 def replace_numbers_with_convert(sentence, wylie=True):
     pattern = r'\d+(\.\d+)?'
     def replace(match):
@@ -21,7 +23,7 @@ converter = pyewts.pyewts()
 checkpoint = "TenzinGayche/TTS_run3_ep20_174k_b"
 processor = SpeechT5Processor.from_pretrained(checkpoint)
 model = SpeechT5ForTextToSpeech.from_pretrained(checkpoint)
-model.to('cuda')
+model.to(device)
 vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan")