add tts device

app.py

@@ -4,12 +4,14 @@ import os
 from datetime import timedelta
 from TTS.api import TTS
 import locale
+import torch
 
 
 locale.getpreferredencoding = lambda: "UTF-8"
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
 os.environ["COQUI_STUDIO_TOS"] = "I AGREE"
-tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2", gpu=True)
+tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2", gpu=True).to(device)
 tts = TTS("xtts_v2.0.2", gpu=True)
 
 
@@ -18,7 +20,7 @@ async def generate_audio(text_input: str, creator: str) -> str:
     refer_voices = ''
     unique_id = str(uuid.uuid4())
 
-    output_file = f'{unique_id}.wav'
+    output_file = f'output.wav'
     match creator:
         case 'Roomie':
             refer_voices = ["./assets/roomie/roomie_emocionado_base_1.wav", "./assets/roomie/ref_12.wav", "./assets/roomie/ref_11.wav", "./assets/roomie/ref_10.wav",