Update app.py

app.py

@@ -31,16 +31,16 @@ tts_model = VitsModel.from_pretrained("facebook/mms-tts-kor")
 tts_tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-kor")
 tts_model.to("cuda" if torch.cuda.is_available() else "cpu")
 
-from uroman import uromanize
+from uroman import Uroman
+
+uroman = Uroman()
 
 def synthesize_tts(text: str):
-    romanized = uromanize(text)
+    romanized = uroman.romanize(text)
     inputs = tts_tokenizer(romanized, return_tensors="pt")
     input_ids = inputs["input_ids"].long().to(tts_model.device)
-
     with torch.no_grad():
         output = tts_model(input_ids=input_ids)
-
     waveform = output.waveform.squeeze().cpu().numpy()
     return (tts_model.config.sampling_rate, waveform)