Update app.py

app.py

@@ -108,11 +108,12 @@ def detect_language(text):
 def generate_audio_kokoro(text, lang):
     """Generate speech using KokoroTTS for supported languages."""
     global kokoro_tts  # Access the preloaded model
-    kokoro_tts.device = "cuda"
+    kokoro_tts.to("cuda")
+    
     lang_code = SUPPORTED_TTS_LANGUAGES.get(lang, "a")  # Default to English
     generator = kokoro_tts(text, voice="af_bella", speed=1, split_pattern=r'\n+')
     # 3. Specify Device
-    device = 'cuda' if torch.cuda.is_available() else 'cpu'
+    #device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
     # Initialize an empty list to store audio data
     audio_data_list = []
@@ -125,7 +126,8 @@ def generate_audio_kokoro(text, lang):
 # Concatenate all audio data into a single array
     full_audio = np.concatenate(audio_data_list)
     # Move model back to CPU after processing
-    kokoro_tts.device = "cpu"
+    #kokoro_tts.device = "cpu"
+    #model.to("cpu")
     
     output_file = f"audio_{lang}.wav"
     sf.write(output_file, full, 24000)  # Save as WAV file