Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ config = AutoConfig.from_pretrained("ckcl/mexc_price_model", config_file_name="s
|
|
25 |
|
26 |
# 创建模型实例并加载权重
|
27 |
model = CustomTransformerModel(config)
|
28 |
-
model.load_state_dict(torch.load("
|
29 |
model.eval() # 设置模型为评估模式
|
30 |
|
31 |
# 定义预测函数
|
|
|
25 |
|
26 |
# 创建模型实例并加载权重
|
27 |
model = CustomTransformerModel(config)
|
28 |
+
model.load_state_dict(torch.load("model_repo/mexc_price.pth"))
|
29 |
model.eval() # 设置模型为评估模式
|
30 |
|
31 |
# 定义预测函数
|