Update app.py

app.py

@@ -64,8 +64,9 @@ chirarity_encoder.fit(torch.range(0, len(CHIRALITY_LIST) - 1).unsqueeze(-1))
 def get_data_list(mol_list):
     data_list = []
     # mol = Chem.MolFromInchi(inchi, sanitize=False, removeHs=False)
-    # mol = Chem.AddHs(mol)
+    # 
     for mol in mol_list:
+        mol = Chem.AddHs(mol)
         weights = []
         type_idx = []
         chirality_idx = []