Update README.md

README.md

@@ -1,6 +1,9 @@
 ---
 license: mit
 library_name: transformers
+tags:
+- image-to-image
+- lineart
 ---
 
 # MangaLineExtraction-hf
@@ -11,7 +14,29 @@ Original repo: https://github.com/ljsabc/MangaLineExtraction_PyTorch
 
 ## Example
 
-TODO
+
+```py
+from PIL import Image
+import torch
+
+from transformers import AutoModel, AutoImageProcessor
+
+REPO_NAME = "p1atdev/MangaLineExtraction-hf"
+
+model = AutoModel.from_pretrained(REPO_NAME, trust_remote_code=True)
+processor = AutoImageProcessor.from_pretrained(REPO_NAME, trust_remote_code=True)
+
+image = Image.open("./sample.jpg")
+
+inputs = processor(image, return_tensors="pt")
+
+with torch.no_grad():
+    outputs = model(inputs.pixel_values)
+
+line_image = Image.fromarray(outputs.pixel_values[0].numpy().astype("uint8"), mode="L")
+line_image.save("./line_image.png")
+```
+
 
 ## Model Details
 
@@ -44,4 +69,4 @@ This is the model card of a 🤗 transformers model that has been pushed on the
     number   = {4},
     pages    = {117:1--117:12},
 }
-```
+```