Manjushri commited on
Commit
f778c96
1 Parent(s): af911a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ pipe = pipe.to(device)
11
  #define interface
12
  def upscale(low_res_img, prompt, negative_prompt, scale, steps):
13
  low_res_img = Image.open(low_res_img).convert("RGB")
14
- low_res_img = low_res_img.resize((256, 256))
15
  upscaled_image = pipe(prompt=prompt, negative_prompt=negative_prompt, image=low_res_img, guidance_scale=scale, num_inference_steps=steps).images[0]
16
  #upscaled_image.save("upsampled.png")
17
  return upscaled_image
 
11
  #define interface
12
  def upscale(low_res_img, prompt, negative_prompt, scale, steps):
13
  low_res_img = Image.open(low_res_img).convert("RGB")
14
+ low_res_img = low_res_img.resize((128, 128))
15
  upscaled_image = pipe(prompt=prompt, negative_prompt=negative_prompt, image=low_res_img, guidance_scale=scale, num_inference_steps=steps).images[0]
16
  #upscaled_image.save("upsampled.png")
17
  return upscaled_image