update app.py

app.py

@@ -105,37 +105,7 @@ def prepare_pipeline(model_version, enable_realism, enable_anti_blur):
     return pipeline
 
 
-@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
-
-
+@spaces.GPU
 def generate_image(
     input_image, 
     control_image, 
@@ -158,8 +128,7 @@ def generate_image(
         seed = torch.seed() & 0xFFFFFFFF
 
     try:
-        image = generate_image_infer(
-            pipeline=pipeline,
+        image = pipeline(
             id_image=input_image,
             prompt=prompt,
             control_image=control_image,