Update run.py

run.py

@@ -54,13 +54,13 @@ def biomarker_prediction(smile_acc, smile_don):
        
             
         das_input =smile_don
-        d_inputs = tokenizer(aas_input, padding='max_length', max_length=400, truncation=True, return_tensors="pt")
+        d_inputs = tokenizer(aas_input, padding='max_length', max_length=510, truncation=True, return_tensors="pt")
         # d_inputs = tokenizer(smiles, truncation=True, return_tensors="pt")
         drug_input_ids = d_inputs['input_ids'].to(device)
         drug_attention_mask = d_inputs['attention_mask'].to(device)
         drug_inputs = {'input_ids': drug_input_ids, 'attention_mask': drug_attention_mask}
 
-        p_inputs = prot_tokenizer(das_input, padding='max_length', max_length=400, truncation=True, return_tensors="pt")
+        p_inputs = prot_tokenizer(das_input, padding='max_length', max_length=510, truncation=True, return_tensors="pt")
         # p_inputs = prot_tokenizer(aas_input, truncation=True, return_tensors="pt")
         prot_input_ids = p_inputs['input_ids'].to(device)
         prot_attention_mask = p_inputs['attention_mask'].to(device)