Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -26,9 +26,9 @@ def upscale_image(model, input_image):
|
|
26 |
|
27 |
# Upscale the image using the selected model
|
28 |
if model == "SD 2.0 2x Latent Upscaler":
|
29 |
-
upscaled_image = sd_2_0_2x(prompt=
|
30 |
else:
|
31 |
-
upscaled_image = sd_2_1_4x(prompt=
|
32 |
|
33 |
# Convert the upscaled tensor back to a PIL image
|
34 |
|
|
|
26 |
|
27 |
# Upscale the image using the selected model
|
28 |
if model == "SD 2.0 2x Latent Upscaler":
|
29 |
+
upscaled_image = sd_2_0_2x(prompt="", image=input_image)
|
30 |
else:
|
31 |
+
upscaled_image = sd_2_1_4x(prompt="", image=input_image)
|
32 |
|
33 |
# Convert the upscaled tensor back to a PIL image
|
34 |
|