Update app.py

app.py

@@ -56,7 +56,7 @@ def generate(
     apply_refiner: bool = False,
     model = 'SG161222/Realistic_Vision_V6.0_B1_noVAE',
     vaecall = 'stabilityai/sd-vae-ft-mse',
-    lora = 'https://m.media-amazon.com/images/I/81zPcrN6m+L.jpg',
+    lora = 'amazonaws-la/juliette',
     lora_scale: float = 0.7,
 ):
     if torch.cuda.is_available():
@@ -72,10 +72,11 @@ def generate(
             pipe.load_lora_weights(lora)
             pipe.fuse_lora(lora_scale=0.7)
             
-        url = lora
+        url = "https://m.media-amazon.com/images/I/81zPcrN6m+L.jpg"
         response = requests.get(url)
         init_image = Image.open(BytesIO(response.content)).convert("RGB")
         init_image = init_image.resize((1024, 1024))
+        init_image.save('source.png')
 
         if ENABLE_CPU_OFFLOAD:
             pipe.enable_model_cpu_offload()