Update app.py
Browse files
app.py
CHANGED
@@ -37,7 +37,7 @@ def load_model():
|
|
37 |
|
38 |
snapshot_download(
|
39 |
repo_id=model_name,
|
40 |
-
use_auth_token=os.environ.get("
|
41 |
allow_patterns=["config.json", "*.safetensors", "model.safetensors.index.json", "vocab.json", "merges.txt", "tokenizer.json"],
|
42 |
ignore_patterns=["optimizer.pt", "pytorch_model.bin", "training_args.bin", "scheduler.pt"]
|
43 |
)
|
@@ -151,9 +151,9 @@ def redistribute_codes(code_list, snac_model):
|
|
151 |
|
152 |
@spaces.GPU()
|
153 |
def generate_speech(text, voice, temperature, top_p, repetition_penalty, max_new_tokens):
|
154 |
-
global tokenizer
|
155 |
-
if tokenizer is None:
|
156 |
-
print("
|
157 |
return None
|
158 |
|
159 |
if not text.strip():
|
|
|
37 |
|
38 |
snapshot_download(
|
39 |
repo_id=model_name,
|
40 |
+
use_auth_token=os.environ.get("HUGGINGFACE_TOKEN"),
|
41 |
allow_patterns=["config.json", "*.safetensors", "model.safetensors.index.json", "vocab.json", "merges.txt", "tokenizer.json"],
|
42 |
ignore_patterns=["optimizer.pt", "pytorch_model.bin", "training_args.bin", "scheduler.pt"]
|
43 |
)
|
|
|
151 |
|
152 |
@spaces.GPU()
|
153 |
def generate_speech(text, voice, temperature, top_p, repetition_penalty, max_new_tokens):
|
154 |
+
global tokenizer, model
|
155 |
+
if tokenizer is None or model is None:
|
156 |
+
print("Model or tokenizer is not initialized. Please ensure the model is properly loaded.")
|
157 |
return None
|
158 |
|
159 |
if not text.strip():
|