Update README.md

README.md

@@ -16,23 +16,38 @@ This model has been pushed to the Hub using the [PytorchModelHubMixin](https://h
 
 Library: [nobg](https://github.com/feyninc/nobg)
 
-## how to load
+## how to use the model
 ```
 pip install nobg
 ```
 
-use the AutoModel class
 ```python
-from nobg import AutoModel
-model = AutoModel.from_pretrained("feyninc/FeyNobg")
-```
-or you can use the model class directly
-```python
-from nobg import BiRefNet
-model = BiRefNet.from_pretrained("feyninc/FeyNobg")
+import torch
+from loadimg import load_img
+from nobg import AutoModel, AutoProcessor
+
+model = AutoModel.from_pretrained("feyninc/FeyNobg").eval()
+processor = AutoProcessor.from_pretrained("feyninc/FeyNobg")
+
+image = load_img("input.jpg").convert("RGB")
+inputs = processor(image, return_tensors="pt")
+
+with torch.no_grad():
+    outputs = model(pixel_values=inputs["pixel_values"])
+
+alpha = processor.post_process_alpha_matting(
+    outputs, target_sizes=[(image.height, image.width)]
+)[0]
+processor.cutout(image, alpha).save("output.png")
 ```
 
 
+![image](https://cdn-uploads.huggingface.co/production/uploads/6527e89a8808d80ccff88b7a/2QIFDxrIJCLlXY87vOzH2.png)
+
+![image](https://cdn-uploads.huggingface.co/production/uploads/6527e89a8808d80ccff88b7a/W0rNUdiP56son6i33vHzY.png)
+
+![image](https://cdn-uploads.huggingface.co/production/uploads/6527e89a8808d80ccff88b7a/ohqZF89NGEm5LXrEOMLP0.png)
+
 ## Citation
 If you use this model, please cite:
 ```bibtex