update
Browse files
app.py
CHANGED
@@ -38,8 +38,8 @@ def generate(dna_sequence, task_type, sample_num=1):
|
|
38 |
Returns:
|
39 |
str: Predicted function label.
|
40 |
"""
|
41 |
-
|
42 |
-
|
43 |
dna_sequence = dna_sequence + task_type +"[MASK]"
|
44 |
|
45 |
tokenized_message = tokenizer(
|
|
|
38 |
Returns:
|
39 |
str: Predicted function label.
|
40 |
"""
|
41 |
+
if task_type is None:
|
42 |
+
task_type = 'H3'
|
43 |
dna_sequence = dna_sequence + task_type +"[MASK]"
|
44 |
|
45 |
tokenized_message = tokenizer(
|