basilboy commited on
Commit
3fffa64
·
verified ·
1 Parent(s): 88dbd92

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +3 -3
model.py CHANGED
@@ -3,7 +3,7 @@
3
  from utils import load_model
4
 
5
  models = {
6
- "hemolysis": load_model("hemolysis_model.pth"),
7
- "non_fouling": load_model("non_fouling_model.pth"),
8
- "solubility": load_model("solubility_model.pth")
9
  }
 
3
  from utils import load_model
4
 
5
  models = {
6
+ "hemolysis": load_model("hemolysis"),
7
+ "non_fouling": load_model("non_fouling"),
8
+ "solubility": load_model("solubility")
9
  }