jasspier commited on
Commit
12ff2bb
1 Parent(s): f7e26e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -25,7 +25,9 @@ torch.hub.download_url_to_file(model_path, 'large.pt')
25
  model = ASRModel()
26
 
27
  # 加载模型参数
28
- model.load_state_dict(torch.load('large.pt', map_location=torch.device('cpu')))
 
 
29
  model.eval()
30
 
31
  # 定义处理函数
 
25
  model = ASRModel()
26
 
27
  # 加载模型参数
28
+ checkpoint = torch.load('large.pt', map_location=torch.device('cpu'))
29
+ state_dict = checkpoint['model'] # 假设模型权重保存在 'model' 键中
30
+ model.load_state_dict(state_dict)
31
  model.eval()
32
 
33
  # 定义处理函数