zehui127 commited on
Commit
26ab4df
·
1 Parent(s): 92ec44f
Files changed (1) hide show
  1. app.py +2 -2
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
- 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(
 
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(