Update app.py
Browse files
app.py
CHANGED
|
@@ -26,12 +26,13 @@ img_pipe.to(device)
|
|
| 26 |
source_img = gr.Image(source="upload", type="filepath", label="init_img | 512*512 px")
|
| 27 |
gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
|
| 28 |
|
| 29 |
-
def resize(
|
| 30 |
-
baseheight =
|
| 31 |
img = Image.open(img)
|
| 32 |
-
hpercent = (baseheight/float(img.size[1]))
|
| 33 |
-
wsize = int((float(img.size[0])*float(hpercent)))
|
| 34 |
-
img = img.resize((wsize,baseheight), Image.Resampling.LANCZOS)
|
|
|
|
| 35 |
return img
|
| 36 |
|
| 37 |
|
|
@@ -54,4 +55,4 @@ print("Great sylvain ! Everything is working fine !")
|
|
| 54 |
title="Img2Img Stable Diffusion CPU"
|
| 55 |
description="Img2Img Stable Diffusion example using CPU and HF token. <br />Warning: Slow process... ~5/10 min inference time. <b>NSFW filter enabled.</b>"
|
| 56 |
|
| 57 |
-
gr.Interface(fn=infer, inputs=["text", source_img], outputs=gallery,title=title,description=description).launch(enable_queue=True)
|
|
|
|
| 26 |
source_img = gr.Image(source="upload", type="filepath", label="init_img | 512*512 px")
|
| 27 |
gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
|
| 28 |
|
| 29 |
+
def resize(value,img):
|
| 30 |
+
#baseheight = value
|
| 31 |
img = Image.open(img)
|
| 32 |
+
#hpercent = (baseheight/float(img.size[1]))
|
| 33 |
+
#wsize = int((float(img.size[0])*float(hpercent)))
|
| 34 |
+
#img = img.resize((wsize,baseheight), Image.Resampling.LANCZOS)
|
| 35 |
+
img = img.resize((value,value), Image.Resampling.LANCZOS)
|
| 36 |
return img
|
| 37 |
|
| 38 |
|
|
|
|
| 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=4).launch(enable_queue=True)
|