Add model card

README.md

@@ -56,7 +56,10 @@ import torch
 
 # Download model
 model_path = hf_hub_download("munchers/scent-to-molecule", "best_control.pt")
-checkpoint = torch.load(model_path, map_location='cpu')
+
+model.load_state_dict(checkpoint['model_state_dict'])
+if torch.cuda.is_available():
+    model = model.cuda()
 ```
 
 ## Examples