BioMike commited on
Commit
fc1023e
1 Parent(s): 5557d6a

Update utils/main_model.py

Browse files
Files changed (1) hide show
  1. utils/main_model.py +1 -0
utils/main_model.py CHANGED
@@ -21,6 +21,7 @@ class ChemicalConverter:
21
  self.iupac_max_len = 156
22
 
23
  def convert(self, input):
 
24
  if self.mode == "SMILES2IUPAC":
25
  tokenizer = self.smiles_tokenizer
26
  reverse_tokenizer = self.iupac_tokenizer
 
21
  self.iupac_max_len = 156
22
 
23
  def convert(self, input):
24
+ input = input.replace(" ", "")
25
  if self.mode == "SMILES2IUPAC":
26
  tokenizer = self.smiles_tokenizer
27
  reverse_tokenizer = self.iupac_tokenizer