Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -23,9 +23,9 @@ def upscale_image(model, input_image):
|
|
23 |
# Upscale the image using the selected model
|
24 |
if model == "SD 2.1 4x Upscaler":
|
25 |
low_res_img = input_image.resize((128, 128))
|
26 |
-
upscaled_image = sd_2_1_4x(prompt="", image=low_res_img, num_inference_steps=5).images[0]
|
27 |
else:
|
28 |
-
upscaled_image = sd_2_0_2x(prompt="", image=input_image, num_inference_steps=5).images[0]
|
29 |
# Return the upscaled image
|
30 |
return upscaled_image
|
31 |
|
|
|
23 |
# Upscale the image using the selected model
|
24 |
if model == "SD 2.1 4x Upscaler":
|
25 |
low_res_img = input_image.resize((128, 128))
|
26 |
+
upscaled_image = sd_2_1_4x(prompt="Sony A7 IV Camera Photo", negative_prompt='Deformity', image=low_res_img, num_inference_steps=5).images[0]
|
27 |
else:
|
28 |
+
upscaled_image = sd_2_0_2x(prompt="Sony A7 IV Camera Photo", , negative_prompt='Deformity', image=input_image, num_inference_steps=5).images[0]
|
29 |
# Return the upscaled image
|
30 |
return upscaled_image
|
31 |
|