Update app.py

app.py

@@ -16,7 +16,7 @@ def apply_patch():
     patch_file = os.path.join(os.path.dirname(__file__), "flux2_klein_kv.patch")
     if os.path.exists(patch_file):
         result = subprocess.run(
-            ["patch", "-p3", "--forward", "--batch"],  # ← p3 strips a/src/diffusers/ prefix
+            ["patch", "-p2", "--forward", "--batch"],  # ← p3 strips a/src/diffusers/ prefix
             cwd=os.path.dirname(site_packages),        # ← cwd is site-packages, not its parent
             stdin=open(patch_file),
             capture_output=True,
@@ -33,7 +33,7 @@ import gradio as gr
 import numpy as np
 import torch
 from PIL import Image
-from diffusers import Flux2KleinKVPipeline
+from diffusers.pipelines.flux2.pipeline_flux2_klein_kv import Flux2KleinKVPipeline
 from huggingface_hub import InferenceClient
 
 dtype = torch.bfloat16