Update app.py

app.py

@@ -4,6 +4,7 @@ import torch
 from transformers.models.speecht5.number_normalizer import EnglishNumberNormalizer
 from string import punctuation
 import re
+import os
 
 
 from parler_tts import ParlerTTSForConditionalGeneration
@@ -14,8 +15,10 @@ device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
 repo_id = "ittailup/la-speech-v0.1"
 
-model = ParlerTTSForConditionalGeneration.from_pretrained(repo_id).to(device)
-tokenizer = AutoTokenizer.from_pretrained(repo_id)
+HF_TOKEN = os.getenv['HF_TOKEN']
+
+model = ParlerTTSForConditionalGeneration.from_pretrained(repo_id, token=HF_TOKEN).to(device)
+tokenizer = AutoTokenizer.from_pretrained(repo_id, token=HF_TOKEN)
 #feature_extractor = AutoFeatureExtractor.from_pretrained(repo_id)