update app.py

app.py

@@ -106,6 +106,36 @@ def prepare_pipeline(model_version, enable_realism, enable_anti_blur):
 
 
 @spaces.GPU(duration=120)
+def generate_image_infer(
+    pipeline,
+    id_image,
+    prompt,
+    control_image,
+    seed,
+    width,
+    height,
+    guidance_scale,
+    num_steps,
+    infusenet_conditioning_scale,
+    infusenet_guidance_start,
+    infusenet_guidance_end,
+):
+    image = pipeline(
+        id_image=id_image,
+        prompt=prompt,
+        control_image=control_image,
+        seed=seed,
+        width=width,
+        height=height,
+        guidance_scale=guidance_scale,
+        num_steps=num_steps,
+        infusenet_conditioning_scale=infusenet_conditioning_scale,
+        infusenet_guidance_start=infusenet_guidance_start,
+        infusenet_guidance_end=infusenet_guidance_end,
+    )
+    return image
+
+
 def generate_image(
     input_image, 
     control_image, 
@@ -128,7 +158,8 @@ def generate_image(
         seed = torch.seed() & 0xFFFFFFFF
 
     try:
-        image = pipeline(
+        image = generate_image_infer(
+            pipeline=pipeline,
             id_image=input_image,
             prompt=prompt,
             control_image=control_image,