Spaces:
Running
on
Zero
Running
on
Zero
Update utils.py
Browse files
utils.py
CHANGED
@@ -240,7 +240,8 @@ class ReactionPredictionModel():
|
|
240 |
pad_token_id=self.tokenizer.pad_token_id,
|
241 |
length_penalty=0.0,
|
242 |
)
|
243 |
-
|
|
|
244 |
original_smiles_list = self.tokenizer.batch_decode(outputs[:, len(inputs['input_ids'][0]):],
|
245 |
skip_special_tokens=True)
|
246 |
original_smiles_list = map(lambda x: x.replace(" ", ""), original_smiles_list)
|
|
|
240 |
pad_token_id=self.tokenizer.pad_token_id,
|
241 |
length_penalty=0.0,
|
242 |
)
|
243 |
+
|
244 |
+
print(outputs)
|
245 |
original_smiles_list = self.tokenizer.batch_decode(outputs[:, len(inputs['input_ids'][0]):],
|
246 |
skip_special_tokens=True)
|
247 |
original_smiles_list = map(lambda x: x.replace(" ", ""), original_smiles_list)
|