Update app.py

app.py

@@ -58,13 +58,9 @@ def generate(
     vaecall = 'stabilityai/sd-vae-ft-mse',
     lora = 'amazonaws-la/juliette',
     lora_scale: float = 0.7,
-):
+    url = "https://m.media-amazon.com/images/I/81zPcrN6m+L.jpg",
+) -> PIL.Image.Image:
     if torch.cuda.is_available():
-
-        url = "https://raw.githubusercontent.com/CompVis/stable-diffusion/main/assets/stable-samples/img2img/sketch-mountains-input.jpg"
-        response = requests.get(url)
-        init_image = Image.open(BytesIO(response.content))
-        init_image = init_image.resize((1024, 1024))
         
         if not use_vae:
             pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model, torch_dtype=torch.float16)
@@ -106,7 +102,7 @@ def generate(
             guidance_scale=guidance_scale_base,
             num_inference_steps=num_inference_steps_base,
             generator=generator,
-            image=init_image,
+            image=url,
             output_type="pil",
         ).images[0]
     else: