Update app.py

app.py

@@ -61,8 +61,12 @@ def generate(
         if not use_vae:  
             pipe = StableDiffusionImg2ImgPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
             response = requests.get(url)
+            if response.status_code != 200:
+            raise Exception(f"Failed to fetch image from URL: {url}")
             init_image = Image.open(BytesIO(response.content)).convert("RGB")
             init_image = init_image.resize((768, 512))
+            except Exception as e:
+            return f"Error fetching image from URL: {str(e)}"
      
         if use_vae:
             vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)