OpenSound commited on
Commit
69e76a4
·
1 Parent(s): 2f0f1b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -139,7 +139,7 @@ def load_models(whisper_backend_name, whisper_model_name, alignment_model_name,
139
  ssrspeech_fn = f"{MODELS_PATH}/{ssrspeech_model_name}.pth"
140
  if not os.path.exists(ssrspeech_fn):
141
  os.system(f"wget https://huggingface.co/westbrook/SSR-Speech-{ssrspeech_model_name}/resolve/main/{ssrspeech_model_name}.pth -O " + ssrspeech_fn)
142
-
143
  ckpt = torch.load(ssrspeech_fn)
144
  model = ssr.SSR_Speech(ckpt["config"])
145
  model.load_state_dict(ckpt["model"])
@@ -442,7 +442,6 @@ def update_demo(mode, smart_transcript, edit_word_mode, transcript, edit_from_wo
442
 
443
 
444
  def get_app():
445
- print('debug4')
446
  with gr.Blocks() as app:
447
  with gr.Row():
448
  with gr.Column(scale=2):
 
139
  ssrspeech_fn = f"{MODELS_PATH}/{ssrspeech_model_name}.pth"
140
  if not os.path.exists(ssrspeech_fn):
141
  os.system(f"wget https://huggingface.co/westbrook/SSR-Speech-{ssrspeech_model_name}/resolve/main/{ssrspeech_model_name}.pth -O " + ssrspeech_fn)
142
+ print(transcribe_model, align_model)
143
  ckpt = torch.load(ssrspeech_fn)
144
  model = ssr.SSR_Speech(ckpt["config"])
145
  model.load_state_dict(ckpt["model"])
 
442
 
443
 
444
  def get_app():
 
445
  with gr.Blocks() as app:
446
  with gr.Row():
447
  with gr.Column(scale=2):