Manjushri commited on
Commit
ae910a4
1 Parent(s): bbca9a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = low_res_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]
 
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]