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

Update model_loader.py

Browse files
Files changed (1) hide show
  1. model_loader.py +3 -1
model_loader.py CHANGED
@@ -5,7 +5,9 @@ import json
5
  from model import LegNet # Import your model
6
 
7
  class CellTypeModelLoader:
8
- def __init__(self, repo_id="Ni-os/Human_Legnet"):
 
 
9
  self.repo_id = repo_id
10
  self.available_cell_types = {
11
  "hepg2": "cell_type_configs/hepg2_config.json",
 
5
  from model import LegNet # Import your model
6
 
7
  class CellTypeModelLoader:
8
+ def __init__(self, repo_id="Ni-os/MPRALegNet
9
+
10
+ "):
11
  self.repo_id = repo_id
12
  self.available_cell_types = {
13
  "hepg2": "cell_type_configs/hepg2_config.json",