Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,9 @@ torch.hub.download_url_to_file(model_path, 'large.pt')
|
|
25 |
model = ASRModel()
|
26 |
|
27 |
# 加载模型参数
|
28 |
-
|
|
|
|
|
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 |
# 定义处理函数
|