test

app.py

@@ -458,9 +458,9 @@ class DreamFuseGUI:
         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)
 
         with gr.Blocks(css=self.css_style) as demo:
             modified_fg_state = gr.State() 
@@ -573,8 +573,8 @@ class DreamFuseGUI:
             )
 
             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