Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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):
|