Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import math  # Added to fix NameError
 import gradio as gr
 import numpy as np
 import random
@@ -66,9 +67,9 @@ pipe = QwenImageEditPlusPipeline.from_pretrained("Qwen/Qwen-Image-Edit-2509",
                                                  scheduler=scheduler,
                                                  torch_dtype=dtype).to(device)
 pipe.load_lora_weights(
-        "lightx2v/Qwen-Image-Lightning", 
-        weight_name="Qwen-Image-Lightning-4steps-V2.0.safetensors"
-    )
+    "lightx2v/Qwen-Image-Lightning", 
+    weight_name="Qwen-Image-Lightning-4steps-V2.0.safetensors"
+)
 pipe.fuse_lora()
 
 # Apply the same optimizations from the first version