Update README.md

README.md

@@ -9,19 +9,26 @@ BEN is a deep learning model designed to automatically remove backgrounds from i
 
 ## Quick Start Code
 ```python
-from BEN import BEN_Base
+from BEN import model
 from PIL import Image
 import torch
 
+
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-model = BEN_Base().to(device).eval()
-model.loadcheckpoints("./BEN/BEN_Base.pth")
 
-image = Image.open("./image2.jpg")
+file = "./image2.jpg" # input image
+
+model = model.BEN_Base().to(device).eval() #init pipeline 
+
+model.loadcheckpoints("./BEN/BEN_Base.pth")
+image = Image.open(file)
 mask, foreground = model.inference(image)
+
 mask.save("./mask.png")
 foreground.save("./foreground.png")
 
+
+
 ```
 # BEN SOA Benchmarks on Disk 5k Eval