Update app.py

app.py

@@ -17,9 +17,10 @@ for model_id in model_ids:
 from TTS.tts.configs.bark_config import BarkConfig
 from TTS.tts.models.bark import Bark
 
-config = BarkConfig()
+#os.environ['CUDA_VISIBLE_DEVICES'] = '1'
+config = BarkConfig(gpu=True)
 model = Bark.init_from_config(config)
-model.load_checkpoint(config, checkpoint_dir="checkpoints/bark", eval=True).to("cuda:0")
+model.load_checkpoint(config, checkpoint_dir="checkpoints/bark", eval=True)
 
 def infer(prompt, input_wav_file):