Update app.py

app.py

@@ -59,14 +59,13 @@ def generate(
 
         if not use_vae:  
             pipe = StableDiffusionImg2ImgPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
+            pipe = pipe.to(device)
+            
             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)