Add image-to-image pipeline tag and license

#1
by nielsr HF Staff - opened
Files changed (1) hide show
  1. README.md +4 -1
README.md CHANGED
@@ -1,5 +1,7 @@
1
  ---
2
  library_name: diffusers
 
 
3
  ---
4
 
5
  # Paint by Inpaint: Learning to Add Image Objects by Removing Them First
@@ -34,6 +36,7 @@ from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDisc
34
  import torch
35
  import requests
36
  from io import BytesIO
 
37
 
38
  model_name = "paint-by-inpaint/general-base"
39
 
@@ -68,4 +71,4 @@ out_images = pipe(
68
  journal={arXiv preprint arXiv:2404.18212},
69
  year={2024}
70
  }
71
-
 
1
  ---
2
  library_name: diffusers
3
+ pipeline_tag: image-to-image
4
+ license: mit
5
  ---
6
 
7
  # Paint by Inpaint: Learning to Add Image Objects by Removing Them First
 
36
  import torch
37
  import requests
38
  from io import BytesIO
39
+ from PIL import Image
40
 
41
  model_name = "paint-by-inpaint/general-base"
42
 
 
71
  journal={arXiv preprint arXiv:2404.18212},
72
  year={2024}
73
  }
74
+ ```