debug

app.py

@@ -20,7 +20,6 @@ vq_model = RQBottleneckTransformer.load_model(
         "whisper-vq-stoks-medium-en+pl-fixed.model"
     ).to(device)
 # tts = TTSProcessor('cpu')
-tts = TTSProcessor("cpu")
 use_8bit = False    
 llm_path = "homebrewltd/Llama3.1-s-instruct-2024-08-19-epoch-3"
 tokenizer = AutoTokenizer.from_pretrained(llm_path)
@@ -72,6 +71,7 @@ def text_to_audio_file(text):
     # remove the last character if it is a period
     if text_split[-1] == ".":
         text_split = text_split[:-1]
+    tts = TTSProcessor("cuda")
     tts.convert_text_to_audio_file(text, temp_file)
     # logging.info(f"Saving audio to {temp_file}")
     # torchaudio.save(temp_file, audio.cpu(), sample_rate=24000)