Update app.py

app.py

@@ -1,10 +1,19 @@
 import gradio as gr
 
-import torch
-from transformers import BarkModel
-from optimum.bettertransformer import BetterTransformer
 
-bark_model = BarkModel.from_pretrained("suno/bark", torch_dtype=torch.float16)
+
+from huggingface_hub import snapshot_download
+
+
+
+
+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}')
+
 
 
 from TTS.tts.configs.bark_config import BarkConfig
@@ -12,7 +21,7 @@ from TTS.tts.models.bark import Bark
 
 config = BarkConfig()
 model = Bark.init_from_config(config)
-model.load_checkpoint(config, checkpoint_dir=bark_model, eval=True)
+model.load_checkpoint(config, checkpoint_dir="checkpoints/bark", eval=True)
 
 def infer(prompt):