Update app.py

app.py

@@ -62,8 +62,11 @@ def generate(
             url = "https://raw.githubusercontent.com/CompVis/stable-diffusion/main/assets/stable-samples/img2img/sketch-mountains-input.jpg"
 
             response = requests.get(url)
+            if response.status_code == 200:
             init_image = Image.open(BytesIO(response.content)).convert("RGB")
             init_image = init_image.resize((768, 512))
+            else:
+            print(f"Error downloading image: {response.status_code}")
      
         if use_vae:
             vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)