Update README.md
#2
by
not-lain
- opened
README.md
CHANGED
@@ -37,6 +37,14 @@ with torch.no_grad():
|
|
37 |
line_image = Image.fromarray(outputs.pixel_values[0].numpy().astype("uint8"), mode="L")
|
38 |
line_image.save("./line_image.png")
|
39 |
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
40 |
|
41 |
|`sample.jpg`|Generated line image|
|
42 |
|-|-|
|
|
|
37 |
line_image = Image.fromarray(outputs.pixel_values[0].numpy().astype("uint8"), mode="L")
|
38 |
line_image.save("./line_image.png")
|
39 |
```
|
40 |
+
or you can use the pipeline
|
41 |
+
|
42 |
+
```py
|
43 |
+
from transformers import pipeline
|
44 |
+
|
45 |
+
pipe = pipeline("image-to-image", model="p1atdev/MangaLineExtraction-hf", trust_remote_code=True)
|
46 |
+
pipe("sample.jpg")
|
47 |
+
```
|
48 |
|
49 |
|`sample.jpg`|Generated line image|
|
50 |
|-|-|
|