Update app.py

app.py

@@ -59,7 +59,6 @@ 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"
 
@@ -93,8 +92,9 @@ def generate(
     if not use_negative_prompt_2:
         negative_prompt_2 = None  # type: ignore
 
-        return pipe(
+        images = pipe(
             prompt=prompt,
+            image=init_image,
             negative_prompt=negative_prompt,
             prompt_2=prompt_2,
             negative_prompt_2=negative_prompt_2,