revise sanitize config and model card usage

Author: liugangcode

torch_molecule/utils/format.py

@@ -58,7 +58,10 @@ def serialize_config(obj):
 
     # Handle numpy arrays
     elif isinstance(obj, (np.ndarray, np.generic)):
-        if obj.size < 1000:
+        # If it's a single number wrapped in a numpy array, just return the number
+        if obj.size == 1:
+            return obj.item()
+        elif obj.size < 1000:
             return {
                 "_type": "numpy_array",
                 "data": obj.tolist(),

torch_molecule/utils/hf.py

@@ -182,11 +182,11 @@ def create_model_card(
     repo="{repo_id}"
 )
 
-# Make predictions
+# For predictor: Make predictions
 # predictions = model.predict(smiles_list)
-# Make generations
+# For generator: Make generations
 # generations = model.generate(n_samples)
-# Make encodings
+# For encoder: Make encodings
 # encodings = model.encode(smiles_list)
 ```