test

.gitattributes

@@ -35,3 +35,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 *.png filter=lfs diff=lfs merge=lfs -text
 *.webp filter=lfs diff=lfs merge=lfs -text
+*.jpg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -46,6 +46,8 @@ class DreamFuseGUI:
             "./examples/030_0.webp"],
             ["./examples/handheld_001_1.png",
             "./examples/handheld_001_0.png"],
+            ["./examples/style_000_1.jpg",
+            "./examples/style_000_0.jpg"],
         ]
         self.examples = [[Image.open(x) for x in example] for example in self.examples]
         self.css_style = self._get_css_style()
@@ -440,9 +442,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() 
@@ -467,32 +469,6 @@ class DreamFuseGUI:
                             draggable_img_in = gr.Image(label="Foreground Image", type="pil", image_mode="RGBA", height=240, width=200)
                     generate_btn = gr.Button("2️⃣ Generate Canvas")
 
-                    with gr.Row():
-                        with gr.Column(scale=1):
-                            gr.Examples(
-                                examples=[self.examples[0]],
-                                inputs=[background_img_in, draggable_img_in],
-                                # elem_id="small-examples"
-                            )
-                        with gr.Column(scale=1):
-                            gr.Examples(
-                                examples=[self.examples[2]],
-                                inputs=[background_img_in, draggable_img_in],
-                                # elem_id="small-examples"
-                            )
-                    with gr.Row():
-                        with gr.Column(scale=1):
-                            gr.Examples(
-                                examples=[self.examples[1]],
-                                inputs=[background_img_in, draggable_img_in],
-                                # elem_id="small-examples"
-                            )
-                        with gr.Column(scale=1):
-                            gr.Examples(
-                                examples=[self.examples[3]],
-                                inputs=[background_img_in, draggable_img_in],
-                                # elem_id="small-examples"
-                            )
                 with gr.Column(scale=1, elem_id="section-preview"):
                     gr.Markdown("### Preview Region")
                     html_out = gr.HTML(
@@ -521,6 +497,32 @@ class DreamFuseGUI:
                     transformation_text = gr.Textbox(label="Transformation Info", elem_id="transformation_info", visible=False)
                     model_output = gr.Image(label="Model Output", type="pil", height=512, width=512)
             
+                with gr.Row():
+                    with gr.Column(scale=1):
+                        gr.Examples(
+                            examples=[self.examples[0]],
+                            inputs=[background_img_in, draggable_img_in],
+                            # elem_id="small-examples"
+                        )
+                    with gr.Column(scale=1):
+                        gr.Examples(
+                            examples=[self.examples[2]],
+                            inputs=[background_img_in, draggable_img_in],
+                            # elem_id="small-examples"
+                        )
+                with gr.Row():
+                    with gr.Column(scale=1):
+                        gr.Examples(
+                            examples=[self.examples[1]],
+                            inputs=[background_img_in, draggable_img_in],
+                            # elem_id="small-examples"
+                        )
+                    with gr.Column(scale=1):
+                        gr.Examples(
+                            examples=[self.examples[3]],
+                            inputs=[background_img_in, draggable_img_in],
+                            # elem_id="small-examples"
+                        )
 
             generate_btn.click(
                 fn=self.on_upload,
@@ -534,8 +536,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