Update app.py

app.py

@@ -50,16 +50,20 @@ def generate(
     num_inference_steps_base: int = 25,
     use_vae: bool = False,
     use_lora: bool = False,
-    model = 'stabilityai/stable-diffusion-xl-base-1.0',
+    model = 'runwayml/stable-diffusion-v1-5',
     vaecall = 'madebyollin/sdxl-vae-fp16-fix',
     lora = '',
+    url = "https://raw.githubusercontent.com/CompVis/stable-diffusion/main/assets/stable-samples/img2img/sketch-mountains-input.jpg",
     lora_scale: float = 0.7,
 ):
     if torch.cuda.is_available():
 
         if not use_vae:  
-            pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
-            
+            pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model, torch_dtype=torch.float16)
+            response = requests.get(url)
+            init_image = Image.open(BytesIO(response.content)).convert("RGB")
+            init_image = init_image.resize((768, 512))
+     
         if use_vae:
             vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)
             pipe = DiffusionPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16)