Ni-os commited on
Commit
a8c4b27
·
verified ·
1 Parent(s): 893acaf

Update model_loader.py

Browse files
Files changed (1) hide show
  1. model_loader.py +2 -2
model_loader.py CHANGED
@@ -87,13 +87,13 @@ class CellTypeModelLoader:
87
  return model
88
 
89
  # Convenience function for easy usage
90
- def load_cell_type_model(cell_type, repo_id="Ni-os/Human_Legnet", **kwargs):
91
  """
92
  Simple function to load model by cell type
93
 
94
  Example:
95
  model = load_cell_type_model("hepg2")
96
- model = load_cell_type_model("k562", repo_id="Ni-os/Human_Legnet")
97
  """
98
  loader = CellTypeModelLoader(repo_id)
99
  return loader.load_model(cell_type, **kwargs)
 
87
  return model
88
 
89
  # Convenience function for easy usage
90
+ def load_cell_type_model(cell_type, repo_id="Ni-os/MPRALegNet", **kwargs):
91
  """
92
  Simple function to load model by cell type
93
 
94
  Example:
95
  model = load_cell_type_model("hepg2")
96
+ model = load_cell_type_model("k562", repo_id="Ni-os/MPRALegNet")
97
  """
98
  loader = CellTypeModelLoader(repo_id)
99
  return loader.load_model(cell_type, **kwargs)