Update app.py

app.py

@@ -349,6 +349,8 @@ def generate_image(
         progress(0.1, desc="Loading LoRA...")
         
         lora = load_file(lora_file)
+        # Move LoRA tensors to CUDA with correct dtype
+        lora = {k: v.to(device="cuda", dtype=torch.bfloat16) for k, v in lora.items()}
         
         progress(0.3, desc="Generating image...")
         
@@ -376,7 +378,7 @@ def create_demo():
     with gr.Blocks(
         title="Z-Image-i2L Demo",
         theme=gr.themes.Soft(),
-        css=".gradio-container { max-width: 1200px !important; }"
+        css=".gradio-container { max-width: 1200px !important; margin: 0 auto}"
     ) as demo:
         gr.Markdown("""
         # 🎨 Z-Image-i2L: Image to LoRA Demo