Update app.py
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ def upscale(raw_img, model, prompt, negative_prompt, scale, steps):
|
|
22 |
generator = torch.manual_seed(999999)
|
23 |
low_res_img = Image.open(raw_img).convert("RGB")
|
24 |
if model == "Upscaler 4x":
|
25 |
-
low_res_img =
|
26 |
image = upscaler4x(prompt=prompt, negative_prompt=negative_prompt, image=low_res_img, guidance_scale=scale, num_inference_steps=steps).images[0]
|
27 |
else:
|
28 |
image = upscaler2x(prompt=prompt, negative_prompt=negative_prompt, image=low_res_img, guidance_scale=scale, num_inference_steps=steps).images[0]
|
|
|
22 |
generator = torch.manual_seed(999999)
|
23 |
low_res_img = Image.open(raw_img).convert("RGB")
|
24 |
if model == "Upscaler 4x":
|
25 |
+
low_res_img = raw_img.resize((128, 128))
|
26 |
image = upscaler4x(prompt=prompt, negative_prompt=negative_prompt, image=low_res_img, guidance_scale=scale, num_inference_steps=steps).images[0]
|
27 |
else:
|
28 |
image = upscaler2x(prompt=prompt, negative_prompt=negative_prompt, image=low_res_img, guidance_scale=scale, num_inference_steps=steps).images[0]
|