michaelapplydesign commited on
Commit
f54cef3
·
1 Parent(s): 656c364
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -71,7 +71,7 @@ def segmentation(img):
71
  def upscale(image,prompt):
72
  print("upscale")
73
 
74
- image = Image.thumbnail((512, 512))
75
 
76
  pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-x4-upscaler")
77
  # pipeline = StableDiffusionLatentUpscalePipeline.from_pretrained("stabilityai/sd-x2-latent-upscaler", torch_dtype=torch.float16).to('cuda')
 
71
  def upscale(image,prompt):
72
  print("upscale")
73
 
74
+ image = image.thumbnail((512, 512))
75
 
76
  pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-x4-upscaler")
77
  # pipeline = StableDiffusionLatentUpscalePipeline.from_pretrained("stabilityai/sd-x2-latent-upscaler", torch_dtype=torch.float16).to('cuda')