Spaces:

Slava917
/

pronunciation-trainer

Runtime error

Slava917 commited on May 17, 2022

Commit

3e9b43d

1 Parent(s): b337855

add eval()

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ torch._C._jit_set_texpr_fuser_enabled(False)
 torch._C._jit_set_nvfuser_enabled(False)
 loader = torch.jit.load("audio_loader.pt")
-model = torch.jit.load('QuartzNet_thunderspeech_3.pt')
 vocab = model.text_transform.vocab.itos
 vocab[-1] = ''

 torch._C._jit_set_nvfuser_enabled(False)
 loader = torch.jit.load("audio_loader.pt")
+model = torch.jit.load('QuartzNet_thunderspeech_3.pt').eval()
 vocab = model.text_transform.vocab.itos
 vocab[-1] = ''