Update README.md

README.md

@@ -6,6 +6,23 @@ license: apache-2.0
 
 BEN is a deep learning model designed to automatically remove backgrounds from images, producing both a mask and a foreground image.
 
+
+## Quick Start Code
+```python
+from BEN import BEN_Base
+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")
+mask, foreground = model.inference(image)
+mask.save("./mask.png")
+foreground.save("./foreground.png")
+
+
 # BEN SOA Benchmarks on Disk 5k Eval 
 
 ### BEN_Base + BEN_Refiner (commercial model please contact us for more information):
@@ -39,17 +56,3 @@ BEN is a deep learning model designed to automatically remove backgrounds from i
 1. Clone Repo
 2. Install requirements.txt
 
-## Quick Start Code
-```python
-from BEN import BEN_Base
-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")
-mask, foreground = model.inference(image)
-mask.save("./mask.png")
-foreground.save("./foreground.png")