Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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"],
|