Update app.py

app.py

@@ -30,12 +30,14 @@ ode_method = "euler"
 def load_model(repo_name, exp_name, model_cls, model_cfg, ckpt_step):
     ckpt_path = str(cached_path(f"hf://SWivid/{repo_name}/{exp_name}/model_{ckpt_step}.safetensors"))
     vocab_char_map, vocab_size = get_tokenizer("Emilia_ZH_EN", "pinyin")
+    
     model = CFM(
         transformer=model_cls(**model_cfg, text_num_embeds=vocab_size, mel_dim=n_mel_channels),
         mel_spec_kwargs=dict(target_sample_rate=target_sample_rate, n_mel_channels=n_mel_channels, hop_length=hop_length),
         odeint_kwargs=dict(method=ode_method),
         vocab_char_map=vocab_char_map,
     ).to(DEVICE)
+    
     model = load_checkpoint(model, ckpt_path, DEVICE, use_ema = True)
     return model