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