amazonaws-sp commited on
Commit
ae00222
·
verified ·
1 Parent(s): e85f8be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -59,7 +59,9 @@ def generate(
59
 
60
  if not use_vae:
61
  pipe = StableDiffusionImg2ImgPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
62
- init_image = Image.open(requests.get("https://raw.githubusercontent.com/CompVis/stable-diffusion/main/assets/stable-samples/img2img/sketch-mountains-input.jpg")).convert("RGB")
 
 
63
  init_image = init_image.resize((768, 512))
64
 
65
  if use_vae:
 
59
 
60
  if not use_vae:
61
  pipe = StableDiffusionImg2ImgPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
62
+ response = requests.get("https://raw.githubusercontent.com/CompVis/stable-diffusion/main/assets/stable-samples/img2img/sketch-mountains-input.jpg")
63
+ image_bytes = BytesIO(response.content)
64
+ init_image = Image.open(image_bytes).convert("RGB")
65
  init_image = init_image.resize((768, 512))
66
 
67
  if use_vae: