Update app.py

app.py

@@ -130,12 +130,6 @@ def load_model(model_type, model_path):
     return model_manager, pipe
 
 
-@spaces.GPU(duration=60)
-def infer(pipe, input_params, seed):
-    torch.manual_seed(seed)
-    return pipe(**input_params)
-
-
 model_dict = {}
 load_model("FLUX", "FLUX.1-dev")
 load_model("Kolors", "Kolors")
@@ -219,6 +213,7 @@ with gr.Blocks() as app:
                     outputs=[output_image],
                     triggers=run_button.click
                 )
+                @spaces.GPU(duration=60)
                 def generate_image(model_type, model_path, prompt, negative_prompt, cfg_scale, embedded_guidance, num_inference_steps, height, width, seed, *args, progress=gr.Progress()):
                     _, pipe = load_model(model_type, model_path)
                     input_params = {
@@ -251,7 +246,8 @@ with gr.Blocks() as app:
                         "masks": masks,
                         "mask_scales": mask_scales,
                     })
-                    image = infer(pipe, input_params, seed)
+                    torch.manual_seed(seed)
+                    image = pipe(**input_params)
                     return image
                 
                 @gr.on(inputs=[output_image] + canvas_list, outputs=canvas_list, triggers=output_to_painter_button.click)