Update app.py
Browse files
app.py
CHANGED
@@ -26,7 +26,7 @@ def upscale(raw_img, model, prompt, negative_prompt, scale, steps):
|
|
26 |
else:
|
27 |
low_res_img = raw_img.resize((128, 128))
|
28 |
upscaled_image = upscaler2x(prompt=prompt, negative_prompt=negative_prompt, image=raw_img, guidance_scale=scale, num_inference_steps=steps).images[0]
|
29 |
-
|
30 |
|
31 |
#launch interface
|
32 |
|
|
|
26 |
else:
|
27 |
low_res_img = raw_img.resize((128, 128))
|
28 |
upscaled_image = upscaler2x(prompt=prompt, negative_prompt=negative_prompt, image=raw_img, guidance_scale=scale, num_inference_steps=steps).images[0]
|
29 |
+
return upscaled_image
|
30 |
|
31 |
#launch interface
|
32 |
|