Create optimization.py

app.py

@@ -8,6 +8,7 @@ import json
 from io import BytesIO
 from PIL import Image, ImageDraw
 from diffusers import Flux2KleinPipeline
+from optimization import optimize_pipeline_
 
 MAX_SEED = np.iinfo(np.int32).max
 
@@ -22,6 +23,16 @@ pipe.load_lora_weights(
     "linoyts/flux_klein_9b_bbox_lora",
     weight_name="flux_klein_9b_bbox_lora.safetensors",
 )
+
+optimize_pipeline_(
+    pipe,
+    image=[Image.new("RGB", (1024, 1024))],
+    prompt = "Move the object inside the red bounding box to the position and size indicated by the green bounding box. Remove the object from its original location in the red box, filling in the background naturally. Remove the bounding boxes and seamlessly blend the repositioned object into the scene, preserving all other objects and the background exactly as they are.",
+    guidance_scale=4,
+    width=1024,
+    height=1024,
+    num_inference_steps=1
+)
 print("Pipeline ready.")
 
 DEFAULT_PROMPT = (