test

app.py

@@ -449,9 +449,9 @@ class DreamblendGUI:
         config = InferenceConfig()
         config.lora_id = 'LL3RD/DreamFuse'
         
-        pipeline = None
-        # pipeline = DreamFuseInference(config)
-        # pipeline.gradio_generate = spaces.GPU(duratioin=120)(pipeline.gradio_generate)
+        # pipeline = None
+        pipeline = DreamFuseInference(config)
+        pipeline.gradio_generate = spaces.GPU(duratioin=120)(pipeline.gradio_generate)
         """创建 Gradio 界面"""
         with gr.Blocks(css=self.css_style) as demo:
             modified_fg_state = gr.State() 
@@ -504,8 +504,8 @@ class DreamblendGUI:
                     cfg_slider = gr.Slider(minimum=1, maximum=10, step=0.1, label="CFG", value=3.5)
                     size_select = gr.Radio(
                         choices=["512", "768", "1024"],
-                        value="512",
-                        label="生成质量(512-差 1024-好)",
+                        value="768",
+                        label="Resolution",
                     )
                     prompt_text = gr.Textbox(label="Prompt", placeholder="text prompt", value="")
                     text_strength = gr.Slider(minimum=1, maximum=10, step=1, label="Text Strength", value=1, visible=False)
@@ -524,8 +524,8 @@ class DreamblendGUI:
                 outputs=[html_out, modified_fg_state],
             )
             model_generate_btn.click(
-                # fn=pipeline.gradio_generate,
-                fn=self.pil_to_base64,
+                fn=pipeline.gradio_generate,
+                # fn=self.pil_to_base64,
                 inputs=[background_img_in, modified_fg_state, transformation_text, seed_slider, \
                     prompt_text, enable_gui, cfg_slider, size_select, text_strength, enable_truecfg],
                 outputs=model_output