From 91e80a5a73da60774605710cdf5628e200a03aed Mon Sep 17 00:00:00 2001
From: Yufan CHEN <108385820+CYF2000127@users.noreply.github.com>
Date: Mon, 12 Aug 2024 22:02:04 +0800
Subject: [PATCH] Update README.md

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 5632291..686dadf 100644
--- a/README.md
+++ b/README.md
@@ -33,7 +33,7 @@ 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)
+predictions = model.predict_final_results(Image)
 print(predictions)
 ```
 or use [`prediction.ipynb`](prediction.ipynb). You can also change the image and model path to your own images and models.