Update app.py

app.py

@@ -61,6 +61,11 @@ def generate(
 ):
     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)
             
@@ -71,11 +76,6 @@ def generate(
         if use_lora:
             pipe.load_lora_weights(lora)
             pipe.fuse_lora(lora_scale=0.7)
-            
-        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 ENABLE_CPU_OFFLOAD:
             pipe.enable_model_cpu_offload()