michaelapplydesign commited on
Commit
e9f68f0
·
1 Parent(s): d51112b

ret 10 images

Browse files
Files changed (1) hide show
  1. app.py +16 -3
app.py CHANGED
@@ -34,13 +34,16 @@ def predict(input_img1,
34
 
35
  print(input_img1, mask,positive_prompt,negative_prompt)
36
 
37
- result_image = make_inpainting(positive_prompt="empty room",
 
 
38
  image=input_img1,
39
  mask_image=mask,
40
  negative_prompt="",
41
  )
 
42
 
43
- return result_image
44
 
45
 
46
  app = gr.Interface(
@@ -48,7 +51,17 @@ app = gr.Interface(
48
  inputs=[gr.Image(label="img", sources=['upload'], type="pil"),
49
  gr.Image(label="mask", sources=['upload'], type="pil")
50
  ],
51
- outputs= gr.Image(label="resp"),
 
 
 
 
 
 
 
 
 
 
52
  title="rem fur 1",
53
  )
54
 
 
34
 
35
  print(input_img1, mask,positive_prompt,negative_prompt)
36
 
37
+ retList=[]
38
+ for x in range(10):
39
+ result_image = make_inpainting(positive_prompt="empty room",
40
  image=input_img1,
41
  mask_image=mask,
42
  negative_prompt="",
43
  )
44
+ retList.append(result_image)
45
 
46
+ return retList
47
 
48
 
49
  app = gr.Interface(
 
51
  inputs=[gr.Image(label="img", sources=['upload'], type="pil"),
52
  gr.Image(label="mask", sources=['upload'], type="pil")
53
  ],
54
+ outputs= [
55
+ gr.Image(label="resp0"),
56
+ gr.Image(label="resp1"),
57
+ gr.Image(label="resp2"),
58
+ gr.Image(label="resp3"),
59
+ gr.Image(label="resp4"),
60
+ gr.Image(label="resp5"),
61
+ gr.Image(label="resp6"),
62
+ gr.Image(label="resp7"),
63
+ gr.Image(label="resp8"),
64
+ gr.Image(label="resp9")],
65
  title="rem fur 1",
66
  )
67