RunningYou commited on
Commit
3183ff3
1 Parent(s): 931a3ea
Files changed (2) hide show
  1. app.py +8 -1
  2. example.png +0 -0
app.py CHANGED
@@ -1,4 +1,3 @@
1
- import imageio
2
  from PIL import Image
3
  import math
4
  import numpy as np
@@ -85,6 +84,14 @@ title = "Person Matting & Stable Diffusion In-Painting"
85
  description = "Inpainting Stable Diffusion <br/><b>mediapipe + Stable Diffusion<b/><br/>"
86
 
87
 
 
 
 
 
 
 
 
 
88
  def predict1(dict, prompt):
89
  dict['image'].save('image.png')
90
  # dict['mask'].save('mask.png')
 
 
1
  from PIL import Image
2
  import math
3
  import numpy as np
 
84
  description = "Inpainting Stable Diffusion <br/><b>mediapipe + Stable Diffusion<b/><br/>"
85
 
86
 
87
+ examples = [
88
+ [
89
+ "a bench in a field",
90
+ 'example.png'
91
+ ]
92
+ ]
93
+
94
+
95
  def predict1(dict, prompt):
96
  dict['image'].save('image.png')
97
  # dict['mask'].save('mask.png')
example.png ADDED