Update app.py
Browse files
app.py
CHANGED
@@ -8,8 +8,8 @@ import numpy as np
|
|
8 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
9 |
|
10 |
# Load model and tokenizer
|
11 |
-
model = ParlerTTSForConditionalGeneration.from_pretrained("TArtx/parler-tts-mini-narrated-
|
12 |
-
tokenizer = AutoTokenizer.from_pretrained("TArtx/parler-tts-mini-narrated-
|
13 |
|
14 |
# Constants
|
15 |
SAMPLE_RATE = model.config.sampling_rate
|
|
|
8 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
9 |
|
10 |
# Load model and tokenizer
|
11 |
+
model = ParlerTTSForConditionalGeneration.from_pretrained("TArtx/parler-tts-mini-narrated-13").to(device)
|
12 |
+
tokenizer = AutoTokenizer.from_pretrained("TArtx/parler-tts-mini-narrated-13")
|
13 |
|
14 |
# Constants
|
15 |
SAMPLE_RATE = model.config.sampling_rate
|