update

app.py

@@ -45,13 +45,13 @@ def main(property: str, smiles: str, smiles_file: str):
     if "Molformer" in property:
         version = property.split(" ")[-1].split("(")[-1].split(")")[0]
         property = property.split(" ")[0]
+        print(property, version)
     algo, config = MOLECULE_PROPERTY_PREDICTOR_FACTORY[property.lower()]
     kwargs = (
         {"algorithm_version": "v0"} if property in MODEL_PROP_DESCRIPTION.keys() else {}
     )
-    if property in MOLFORMER_VERSIONS.keys():
+    if property.lower() in MOLFORMER_VERSIONS.keys():
         kwargs["algorithm_version"] = version
-
     model = algo(config(**kwargs))
     if smiles is not None and smiles_file is not None:
         raise ValueError("Pass either smiles or smiles_file, not both.")