lgq12697 commited on
Commit
6d92ac5
·
verified ·
1 Parent(s): aa66fb7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -5,6 +5,8 @@ import torch.nn.functional as F
5
  placeholder = 'TTCATCTCGTCCGACGCTTCAACCCGCACCGATCCTGCGCCACCCCTTCGCCGGCGGCTTCTCCCCTCCTCTTCCTCCGCCGCTGCATCGCCGTCCCAGGAACTTGGACACGTCGCCTCTCGCCGGCGACCATGTACCGCGCCCTCCGCTCTCTCAAGGTTTCCCCGTCTGCACCCCCCCAACCTTCTACGACGTGTGGCGTTGCGTGTCTCGATCCATTTGGGATGAATGCGCTGGAGTGTTAGA'
6
  model_names = ['plant-dnabert', 'plant-dnagpt', 'plant-nucleotide-transformer', 'plant-dnagemma',
7
  'dnabert2', 'nucleotide-transformer-v2-100m', 'agront-1b']
 
 
8
  task_map = {
9
  "promoter": ["Not promoter", "Core promoter"],
10
  "conservation": ["Not conserved", "Conserved"],
 
5
  placeholder = 'TTCATCTCGTCCGACGCTTCAACCCGCACCGATCCTGCGCCACCCCTTCGCCGGCGGCTTCTCCCCTCCTCTTCCTCCGCCGCTGCATCGCCGTCCCAGGAACTTGGACACGTCGCCTCTCGCCGGCGACCATGTACCGCGCCCTCCGCTCTCTCAAGGTTTCCCCGTCTGCACCCCCCCAACCTTCTACGACGTGTGGCGTTGCGTGTCTCGATCCATTTGGGATGAATGCGCTGGAGTGTTAGA'
6
  model_names = ['plant-dnabert', 'plant-dnagpt', 'plant-nucleotide-transformer', 'plant-dnagemma',
7
  'dnabert2', 'nucleotide-transformer-v2-100m', 'agront-1b']
8
+ tokenizer_type = "BPE"
9
+ model_names = [x + '-' + tokenizer_type if x.startswith("plant") else x for x in model_names]
10
  task_map = {
11
  "promoter": ["Not promoter", "Core promoter"],
12
  "conservation": ["Not conserved", "Conserved"],