Update app.py

app.py

@@ -47,6 +47,9 @@ def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora
         joint_attention_kwargs={"scale": lora_scale},
     ).images[0]
 
+    # Save image using the pipeline
+    image.save('/images_generated/'+prompt+'.webp')
+
     # Final update (100%)
     progress(100, "Completed!")