update
Browse files
app.py
CHANGED
@@ -38,7 +38,10 @@ def generate(dna_sequence, task_type, sample_num=1):
|
|
38 |
Returns:
|
39 |
str: Predicted function label.
|
40 |
"""
|
|
|
|
|
41 |
dna_sequence = dna_sequence + task_type +"[MASK]"
|
|
|
42 |
tokenized_message = tokenizer(
|
43 |
[dna_sequence], return_tensors='pt', return_token_type_ids=False, add_special_tokens=True
|
44 |
).to(device)
|
|
|
38 |
Returns:
|
39 |
str: Predicted function label.
|
40 |
"""
|
41 |
+
print(dna_sequence,type(dna_sequence))
|
42 |
+
print(task_type,type(task_type))
|
43 |
dna_sequence = dna_sequence + task_type +"[MASK]"
|
44 |
+
|
45 |
tokenized_message = tokenizer(
|
46 |
[dna_sequence], return_tensors='pt', return_token_type_ids=False, add_special_tokens=True
|
47 |
).to(device)
|