update

app.py

@@ -41,19 +41,16 @@ MODEL_PROP_DESCRIPTION = {
 
 
 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.lower() in MOLFORMER_VERSIONS.keys():
         kwargs["algorithm_version"] = version
-    print(property.lower(), MOLFORMER_VERSIONS.keys())
-    print("KWARGS", kwargs)
+
     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.")
@@ -78,7 +75,6 @@ def main(property: str, smiles: str, smiles_file: str):
 
 
 if __name__ == "__main__":
-
     # Preparation (retrieve all available algorithms)
     properties = list(MOLECULE_PROPERTY_PREDICTOR_FACTORY.keys())[::-1]
     for prop in REMOVE:

utils.py

@@ -24,6 +24,7 @@ def draw_grid_predict(
     Returns:
         HTML to display
     """
+    print("Mols2Grid", mols2grid.__version__)
 
     if domain not in ["Molecules", "Proteins"]:
         raise ValueError(f"Unsupported domain {domain}")