Update app.py

app.py

@@ -6,17 +6,6 @@ login(token=os.getenv('HF_AK'))
 from diffsynth import download_models
 download_models(["Kolors", "FLUX.1-dev"], downloading_priority=["HuggingFace", "ModelScope"])
 
-def get_file_list(path):
-    file_list = []
-    for filename in os.listdir(path):
-        file_path = os.path.join(path, filename)
-        if os.path.isdir(file_path):
-            file_list.extend(get_file_list(file_path))
-        else:
-            file_list.append(file_path)
-    return file_list
-print([i for i in get_file_list("models") if "cache" not in i])
-
 import gradio as gr
 from diffsynth import ModelManager, SDImagePipeline, SDXLImagePipeline, SD3ImagePipeline, HunyuanDiTImagePipeline, FluxImagePipeline
 import os, torch
@@ -141,6 +130,12 @@ 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 = {}
 
 with gr.Blocks() as app:
@@ -222,7 +217,6 @@ 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 = {
@@ -255,8 +249,7 @@ with gr.Blocks() as app:
                         "masks": masks,
                         "mask_scales": mask_scales,
                     })
-                    torch.manual_seed(seed)
-                    image = pipe(**input_params)
+                    image = infer(pipe, input_params, seed)
                     return image
                 
                 @gr.on(inputs=[output_image] + canvas_list, outputs=canvas_list, triggers=output_to_painter_button.click)