Update app.py

app.py

@@ -3,12 +3,12 @@ import torch
 from transformers import BarkModel
 from optimum.bettertransformer import BetterTransformer
 
-model = BarkModel.from_pretrained("suno/bark-small", torch_dtype=torch.float16)
+model = BarkModel.from_pretrained("suno/bark", torch_dtype=torch.float16)
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 model = model.to(device)
 
 from transformers import AutoProcessor
-processor = AutoProcessor.from_pretrained("suno/bark-small")
+processor = AutoProcessor.from_pretrained("suno/bark")
 
 # Use bettertransform for flash attention
 model = BetterTransformer.transform(model, keep_original_model=False)