Update README.md

CYF2000127 · Aug 12, 2024 · bd5f05e · bd5f05e
1 parent 55bac6b
commit bd5f05e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -29,11 +29,11 @@ pip install -r requirements.txt
 ```python
 import torch
 from MolNexTR import molnextr
-IMAGE = './examples/1.png'
-MODEL = './checkpoints/molnextr_best.pth'
+Image = './examples/1.png'
+Model = './checkpoints/molnextr_best.pth'
 device = torch.device('cpu')
-model = molnextr(MODEL, device)
-predictions = model.predict_final_results(IMAGE, return_atoms_bonds=True)
+model = molnextr(Model, device)
+predictions = model.predict_final_results(Image, return_atoms_bonds=True)
 print(predictions)
 ```
 or use [`prediction.ipynb`](prediction.ipynb). You can also change the image and model path to your own images and models.