Update code snippet (#11)

- Update code snippet (12143b7b7ed84c60db6f1df51f6392d557101264)


Co-authored-by: Niels Rogge <[email protected]>

README.md

@@ -107,7 +107,6 @@ import torch, os
 from PIL import Image
 from briarmbg import BriaRMBG
 from utilities import preprocess_image, postprocess_image
-from huggingface_hub import hf_hub_download
 
 im_path = f"{os.path.dirname(os.path.abspath(__file__))}/example_input.jpg"
 
@@ -115,7 +114,6 @@ net = BriaRMBG()
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 net = BriaRMBG.from_pretrained("briaai/RMBG-1.4")
 net.to(device)
-net.eval()    
 
 # prepare input
 model_input_size = [1024,1024]