Skip to content

Commit

Permalink
Replace models, change dependencies
Browse files Browse the repository at this point in the history
  • Loading branch information
rodvrees committed Sep 14, 2024
1 parent e014cec commit da9e4f5
Show file tree
Hide file tree
Showing 6 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion im2deep/im2deep.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ def predict_ccs(
if model_name == "tims":
path_model = Path(__file__).parent / "models" / "TIMS"

path_model_list = list(path_model.glob("*.hdf5"))
path_model_list = list(path_model.glob("*.keras"))
if use_single_model:
LOGGER.debug("Using model {}".format(path_model_list[2]))
path_model_list = [path_model_list[2]]
Expand Down
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ dynamic = ["version"]
requires-python = ">=3.8"
dependencies = [
"click",
"deeplc==2.2.38",
"deeplc",
"psm_utils",
"pandas",
"numpy",
Expand Down

0 comments on commit da9e4f5

Please sign in to comment.