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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,12 +20,12 @@ upscaler4x = upscaler4x.to(device)
20
 
21
  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 = 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]
29
  return image
30
 
31
  #launch interface
 
20
 
21
  def upscale(raw_img, model, prompt, negative_prompt, scale, steps):
22
  generator = torch.manual_seed(999999)
23
+ raw_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=raw_img, guidance_scale=scale, num_inference_steps=steps).images[0]
29
  return image
30
 
31
  #launch interface