We expanded our testing to the PubChem dataset to explore the model's capabilities across varied molecular weights and chemical structures beyond the initial training set. We curated three sets of 100 molecules from PubChem, categorized into molecular weight ranges: 0-250 Da (see Figure ), 250-350 Da, and 350-500 Da (see Supplementary Figure ).