Hugging Face's logo

Spaces:
yonghan93
/
CVD
Sleeping

App Files
yonghan93 commited on
Commit
7f36c84
·
verified ·
1 Parent(s): 36a175c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import pickle
7
  from pytorch_tabular import TabularModel
8
 
9
  # Load model
10
- model = TabularModel.load("FTTransformerModel")
11
 
12
  # Load threshold
13
  with open("FTTransformerModel/threshold.json", "r") as f:
 
7
  from pytorch_tabular import TabularModel
8
 
9
  # Load model
10
+ model = TabularModel.load_model("FTTransformerModel")
11
 
12
  # Load threshold
13
  with open("FTTransformerModel/threshold.json", "r") as f: