Update app.py

app.py

@@ -9,6 +9,7 @@ from scipy.io import wavfile
 model_ids = [
     'suno/bark',
 ]
+
 for model_id in model_ids:
     model_name = model_id.split('/')[-1]
     snapshot_download(model_id, local_dir=f'checkpoints/{model_name}')
@@ -18,7 +19,7 @@ from TTS.tts.models.bark import Bark
 
 config = BarkConfig()
 model = Bark.init_from_config(config)
-model.load_checkpoint(config, checkpoint_dir="checkpoints/bark", eval=True)
+model.load_checkpoint(config, checkpoint_dir="checkpoints/bark", eval=True).to("cuda:0")
 
 def infer(prompt, input_wav_file):
 
@@ -116,6 +117,7 @@ def infer_with_npz(prompt, input_wav_file):
     )
 
     # Print again the contents
+    contents = os.listdir(f"bark_voices/{file_name}")
     for item in contents:
         print(item)