bluenevus commited on
Commit
122b144
·
verified ·
1 Parent(s): 6dcce28

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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("HF_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,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("Tokenizer is not initialized. Please ensure the model is properly loaded.")
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():