ckcl commited on
Commit
66a9d1c
·
verified ·
1 Parent(s): 637f170

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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("ckcl/mexc_price.pth"))
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
  # 定义预测函数