fffiloni commited on
Commit
c4bb4d9
1 Parent(s): e3f2186

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -36,7 +36,7 @@ def resize(value,img):
36
  return img
37
 
38
 
39
- def infer(prompt, source_img):
40
 
41
  source_image = resize(512, source_img)
42
  source_image.save('source.png')
@@ -55,4 +55,4 @@ print("Great sylvain ! Everything is working fine !")
55
  title="Img2Img Stable Diffusion CPU"
56
  description="Img2Img Stable Diffusion example using CPU and HF token. <br />Warning: Slow process... ~5/10 min inference time. <b>NSFW filter enabled.</b>"
57
 
58
- gr.Interface(fn=infer, inputs=["text", source_img], outputs=gallery,title=title,description=description).queue(max_size=100).launch(enable_queue=True)
 
36
  return img
37
 
38
 
39
+ def infer(source_img, prompt):
40
 
41
  source_image = resize(512, source_img)
42
  source_image.save('source.png')
 
55
  title="Img2Img Stable Diffusion CPU"
56
  description="Img2Img Stable Diffusion example using CPU and HF token. <br />Warning: Slow process... ~5/10 min inference time. <b>NSFW filter enabled.</b>"
57
 
58
+ gr.Interface(fn=infer, inputs=[source_img, "text"], outputs=gallery,title=title,description=description).queue(max_size=100).launch(enable_queue=True)