Upload model card

README.md

@@ -36,13 +36,12 @@ pip install como-ocsr
 ## Usage
 
 ```python
-from como.model import ComoModel
-model = ComoModel.from_pretrained("Keylab/COMO-Tanimoto", device="cuda")
-
-# or download checkpoint from HuggingFace:
-# huggingface-cli download Keylab/COMO models/tanimoto/final.pth
 import como
-model = como.load_model("models/tanimoto/final.pth", device="cuda")
+# Load pretrained checkpoints from Hugging Face
+model = como.load_model("Keylab/COMO-Tanimoto")
+# or download checkpoint files from HuggingFace:
+# huggingface-cli download Keylab/COMO models/tanimoto/final.pth
+model = como.load_model(checkpint_path="models/tanimoto/final.pth")
 
 # Single image prediction
 smiles = como.predict(model, "molecule.png")