Update app.py

app.py

@@ -18,7 +18,7 @@ from TTS.tts.configs.bark_config import BarkConfig
 from TTS.tts.models.bark import Bark
 
 #os.environ['CUDA_VISIBLE_DEVICES'] = '1'
-config = BarkConfig(gpu=True)
+config = BarkConfig()
 model = Bark.init_from_config(config)
 model.load_checkpoint(config, checkpoint_dir="checkpoints/bark", eval=True)
 
@@ -56,7 +56,8 @@ def infer(prompt, input_wav_file):
         text, 
         config, 
         speaker_id=f"{file_name}", 
-        voice_dirs="bark_voices/"
+        voice_dirs="bark_voices/",
+        gpu=True
     )
     
     print(output_dict)