Update README.md

README.md

@@ -33,8 +33,7 @@ model = model.BEN_Base().to(device).eval() #init pipeline
 
 model.loadcheckpoints("./BEN_Base.pth")
 image = Image.open(file)
-with torch.no_grad():
-    mask, foreground = model.inference(image)
+mask, foreground = model.inference(image)
 
 mask.save("./mask.png")
 foreground.save("./foreground.png")