Update app.py

app.py

@@ -59,10 +59,13 @@ def generate(
 
         if not use_vae:  
             pipe = StableDiffusionImg2ImgPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
-            response = requests.get("https://raw.githubusercontent.com/CompVis/stable-diffusion/main/assets/stable-samples/img2img/sketch-mountains-input.jpg")
-            image_bytes = BytesIO(response.content)
-            init_image = Image.open(image_bytes).convert("RGB")
-            init_image = init_image.resize((768, 512))
+            img = "https://raw.githubusercontent.com/CompVis/stable-diffusion/main/assets/stable-samples/img2img/sketch-mountains-input.jpg"
+            def resize(value,img):
+            img = Image.open(img)
+            img = img.resize((value,value))
+            return img
+            source_image = resize(768, source_img)
+            source_image.save('source.png')
      
         if use_vae:
             vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)