20240706

app.py

@@ -184,18 +184,14 @@ def demo_openlrm(infer_impl):
             gr.Markdown(
                 """
                 ## Example in the paper.
-                ### A. 3D Style Transfer
-                Here we keep the object ID and just transfer the style. <br>
-
-                **Line 1: A pop-mart boy with astronaut, blue, traditional Chinese and grey style.**
                 """
             )
         with gr.Row():
             examples = [
                 ['assets/sample_input/demo/front/boy_astronaut.png', 'assets/sample_input/demo/back/boy_astronaut.png'],
-                ['assets/sample_input/demo/front/boy_blue.png', 'assets/sample_input/demo/back/boy_blue.png'],
                 ['assets/sample_input/demo/front/boy_chinese_style.png', 'assets/sample_input/demo/back/boy_chinese_style.png'],
-                ['assets/sample_input/demo/front/boy_grey_clothes.png', 'assets/sample_input/demo/back/boy_grey_clothes.png'],
+                ['assets/sample_input/demo/front/kunkun_soccer.png', 'assets/sample_input/demo/back/kunkun_soccer.png'],
+                ['assets/sample_input/demo/front/marvel_superman.png', 'assets/sample_input/demo/back/marvel_superman.png'],
             ]
 
             for example in examples:
@@ -209,138 +205,25 @@ def demo_openlrm(infer_impl):
                         examples_per_page=3,
                     )
 
-        # # EXAMPLES
-        # with gr.Row():
-        #     gr.Markdown(
-        #         """
-        #         **Line 2: A LEGO model featuring an astronaut, green and red elements, and a wizard theme.**
-        #         """
-        #     )
-        # with gr.Row():
-        #     examples = [
-        #         ['assets/sample_input/demo/front/lego_astronaut.png', 'assets/sample_input/demo/back/lego_astronaut.png'],
-        #         ['assets/sample_input/demo/front/lego_green.png', 'assets/sample_input/demo/back/lego_green.png'],
-        #         ['assets/sample_input/demo/front/lego_red.png', 'assets/sample_input/demo/front/lego_red.png'],
-        #         ['assets/sample_input/demo/front/lego_wizard.png', 'assets/sample_input/demo/back/lego_wizard.png'],
-        #     ]
-
-        #     for example in examples:
-        #         with gr.Column(scale=0.3):
-        #             gr.Examples(
-        #                 examples=[example],
-        #                 inputs=[input_front_image, input_back_image], 
-        #                 outputs=None, # [processed_image, output_video],
-        #                 fn=None, # example_fn,
-        #                 cache_examples=bool(os.getenv('SPACE_ID')),
-        #                 examples_per_page=3,
-        #             )
-        # with gr.Row():
-        #     gr.Markdown(
-        #         """
-        #         **Line 3: A marvel boy featuring an Captain America, Ironman and Spiderman, and a Superman theme.**
-        #         """
-        #     )
-        # with gr.Row():
-        #     examples = [
-        #         ['assets/sample_input/demo/front/marvel_captain.png', 'assets/sample_input/demo/back/marvel_captain.png'],
-        #         ['assets/sample_input/demo/front/marvel_ironman.png', 'assets/sample_input/demo/front/marvel_ironman.png'],
-        #         ['assets/sample_input/demo/front/marvel_spiderman.png', 'assets/sample_input/demo/back/marvel_spiderman.png'],
-        #         ['assets/sample_input/demo/front/marvel_superman.png', 'assets/sample_input/demo/back/marvel_superman.png'],
-        #     ]
-
-        #     for example in examples:
-        #         with gr.Column(scale=0.3):
-        #             gr.Examples(
-        #                 examples=[example],
-        #                 inputs=[input_front_image, input_back_image], 
-        #                 outputs=None, # [processed_image, output_video],
-        #                 fn=None, # example_fn,
-        #                 cache_examples=bool(os.getenv('SPACE_ID')),
-        #                 examples_per_page=3,
-        #             )
-        # # EXAMPLES
-        # with gr.Row():
-        #     gr.Markdown(
-        #         """
-        #         ### B. 3D Generative Geometry or Pattern Fill
-                
-        #         Here, we start with a simple object and gradually add various accessories, costumes, or patterns step by step. We only showcase the final effect after multiple rounds of decoration. <br>
-
-        #         **Line 4: Initial object: sofa, dog, penguin, house.**
-        #         """
-        #     )
-        # with gr.Row():
-        #     examples = [
-        #         ['assets/sample_input/demo/front/sofa.png', 'assets/sample_input/demo/back/sofa.png'],
-        #         ['assets/sample_input/demo/front/penguin.png', 'assets/sample_input/demo/back/penguin.png'],
-        #         ['assets/sample_input/demo/front/house.png', 'assets/sample_input/demo/back/house.png'],
-        #     ]
-
-        #     for example in examples:
-        #         with gr.Column(scale=0.3):
-        #             gr.Examples(
-        #                 examples=[example],
-        #                 inputs=[input_front_image, input_back_image], 
-        #                 outputs=None, # [processed_image, output_video],
-        #                 fn=None, # example_fn,
-        #                 cache_examples=bool(os.getenv('SPACE_ID')),
-        #                 examples_per_page=3,
-        #             )
-
-        # with gr.Row():
-        #     gr.Markdown(
-        #         """
-        #         ### C. 3D Style Fusion
-                
-        #         We will maintain a consistent front style of the object while continuously changing the back style, blending the two different styles into one object.<br>
-
-        #         **Line 5: A bird with different back styles.**
-        #         """
-        #     )
-        # with gr.Row():
-        #     examples = [
-        #         ['assets/sample_input/demo/front/bird.png', 'assets/sample_input/demo/back/bird.png'],
-        #         ['assets/sample_input/demo/front/bird_brownblue.png', 'assets/sample_input/demo/back/bird_brownblue.png'],
-        #         ['assets/sample_input/demo/front/bird_rainbow.png', 'assets/sample_input/demo/back/bird_rainbow.png'],
-        #         ['assets/sample_input/demo/front/bird_whitered.png', 'assets/sample_input/demo/back/bird_whitered.png'],
-        #     ]
-
-        #     for example in examples:
-        #         with gr.Column(scale=0.3):
-        #             gr.Examples(
-        #                 examples=[example],
-        #                 inputs=[input_front_image, input_back_image], 
-        #                 outputs=None, # [processed_image, output_video],
-        #                 fn=None, # example_fn,
-        #                 cache_examples=bool(os.getenv('SPACE_ID')),
-        #                 examples_per_page=3,
-        #             )
-
-        # with gr.Row():
-        #     gr.Markdown(
-        #         """
-        #         ### Others
-        #         I vote for kunkun forever, I am really an I-kUN and have heard many of his songs.
-        #         """
-        #     )
-        # with gr.Row():
-        #     examples = [
-        #         ['assets/sample_input/demo/front/loopy.png', 'assets/sample_input/demo/back/loopy.png'],
-        #         ['assets/sample_input/demo/front/mario.png', 'assets/sample_input/demo/back/mario.png'],
-        #         ['assets/sample_input/demo/front/armor.png', 'assets/sample_input/demo/back/armor.png'],
-        #         ['assets/sample_input/demo/front/kunkun_law.png', 'assets/sample_input/demo/back/kunkun_law.png'],
-        #     ]
-
-        #     for example in examples:
-        #         with gr.Column(scale=0.3):
-        #             gr.Examples(
-        #                 examples=[example],
-        #                 inputs=[input_front_image, input_back_image], 
-        #                 outputs=None, # [processed_image, output_video],
-        #                 fn=None, # example_fn,
-        #                 cache_examples=bool(os.getenv('SPACE_ID')),
-        #                 examples_per_page=3,
-        #             )
+        # EXAMPLES
+        with gr.Row():
+            examples = [
+                ['assets/sample_input/demo/front/lego_astronaut.png', 'assets/sample_input/demo/back/lego_astronaut.png'],
+                ['assets/sample_input/demo/front/lego_wizard.png', 'assets/sample_input/demo/back/lego_wizard.png'],
+                ['assets/sample_input/demo/front/bird.png', 'assets/sample_input/demo/back/bird.png'],
+                ['assets/sample_input/demo/front/bird_whitered.png', 'assets/sample_input/demo/back/bird_whitered.png'],
+            ]
+
+            for example in examples:
+                with gr.Column(scale=0.3):
+                    gr.Examples(
+                        examples=[example],
+                        inputs=[input_front_image, input_back_image], 
+                        outputs=None, # [processed_image, output_video],
+                        fn=None, # example_fn,
+                        cache_examples=bool(os.getenv('SPACE_ID')),
+                        examples_per_page=3,
+                    )
 
         working_dir = gr.State()
         submit.click(