Spaces:
Running
on
Zero
Running
on
Zero
update
Browse files- upscale_image.py +3 -2
upscale_image.py
CHANGED
@@ -17,14 +17,15 @@ def execute(input_image):
|
|
17 |
pipe.vae.enable_tiling()
|
18 |
# 画像のパスとプロンプト
|
19 |
prompt = "beautiful girl"
|
20 |
-
first_resize_w =
|
21 |
-
first_resize_h =
|
22 |
|
23 |
# 画像の読み込みとリサイズ
|
24 |
image = input_image#.convert("RGB")
|
25 |
low_res_img = image
|
26 |
if first_resize_w!=0 and first_resize_h!=0:
|
27 |
low_res_img = image.resize((first_resize_w, first_resize_h))
|
|
|
28 |
upscaled_image = upscale(pipe, prompt, low_res_img)
|
29 |
return upscaled_image
|
30 |
|
|
|
17 |
pipe.vae.enable_tiling()
|
18 |
# 画像のパスとプロンプト
|
19 |
prompt = "beautiful girl"
|
20 |
+
first_resize_w = 64
|
21 |
+
first_resize_h = 64
|
22 |
|
23 |
# 画像の読み込みとリサイズ
|
24 |
image = input_image#.convert("RGB")
|
25 |
low_res_img = image
|
26 |
if first_resize_w!=0 and first_resize_h!=0:
|
27 |
low_res_img = image.resize((first_resize_w, first_resize_h))
|
28 |
+
|
29 |
upscaled_image = upscale(pipe, prompt, low_res_img)
|
30 |
return upscaled_image
|
31 |
|