bernardo-de-almeida commited on
Commit
7163437
·
1 Parent(s): 549ec02

feat: extract DNA sequence instead of downloading genome

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -8,11 +8,17 @@ from ntv3_tracks_pipeline import load_ntv3_tracks_pipeline
8
  MODEL_ID = os.environ.get("MODEL_ID", "InstaDeepAI/NTv3_650M_pos")
9
  DEFAULT_SPECIES = os.environ.get("DEFAULT_SPECIES", "human")
10
 
 
 
 
 
 
11
  # Load once at startup (Space container)
12
  pipe = load_ntv3_tracks_pipeline(
13
  model=MODEL_ID,
14
  device="auto",
15
  default_species=DEFAULT_SPECIES,
 
16
  verbose=False,
17
  )
18
 
 
8
  MODEL_ID = os.environ.get("MODEL_ID", "InstaDeepAI/NTv3_650M_pos")
9
  DEFAULT_SPECIES = os.environ.get("DEFAULT_SPECIES", "human")
10
 
11
+ HF_TOKEN = (
12
+ os.environ.get("HF_TOKEN")
13
+ or os.environ.get("HUGGINGFACEHUB_API_TOKEN") # also common in Spaces
14
+ )
15
+
16
  # Load once at startup (Space container)
17
  pipe = load_ntv3_tracks_pipeline(
18
  model=MODEL_ID,
19
  device="auto",
20
  default_species=DEFAULT_SPECIES,
21
+ token=HF_TOKEN,
22
  verbose=False,
23
  )
24