fix example argument

app_fatezero.py

@@ -212,9 +212,9 @@ with gr.Blocks(css='style.css') as demo:
             # - It takes a few minutes to download model first.
             # - Expected time to generate an 8-frame video: 70 seconds with T4, 24 seconds with A10G, (10 seconds with A100)
             # ''')
-            gr.Markdown('''
-            todo
-            ''')
+            # gr.Markdown('''
+            # todo
+            # ''')
         with gr.Column():
             result = gr.Video(label='Result')
             result.style(height=512, width=512)
@@ -246,32 +246,34 @@ with gr.Blocks(css='style.css') as demo:
                                 step=1,
                                 value=10)
     with gr.Row():
-        examples = [
-            [
-                'CompVis/stable-diffusion-v1-4',
-                'FateZero/data/teaser_car-turn',
-                'a silver jeep driving down a curvy road in the countryside',
-                'watercolor painting of a silver jeep driving down a curvy road in the countryside',
-                0.8, 
-                0.8,
-                "watercolor",
-                10,
-                10,
-                7.5,
-            ],
-            [
-                'CompVis/stable-diffusion-v1-4',
-                'FateZero/data/style/sunflower',
-                'a yellow sunflower',
-                'van gogh style painting of a yellow sunflower',
-                0.5,
-                0.5,
-                'van gogh',
-                10,
-                10,
-                7.5,
-            ],
-        ]
+        from example import style_example
+        examples = style_example
+        # examples = [
+        #     [
+        #         'CompVis/stable-diffusion-v1-4',
+        #         'FateZero/data/teaser_car-turn',
+        #         'a silver jeep driving down a curvy road in the countryside',
+        #         'watercolor painting of a silver jeep driving down a curvy road in the countryside',
+        #         0.8, 
+        #         0.8,
+        #         "watercolor",
+        #         10,
+        #         10,
+        #         7.5,
+        #     ],
+        #     [
+        #         'CompVis/stable-diffusion-v1-4',
+        #         'FateZero/data/style/sunflower',
+        #         'a yellow sunflower',
+        #         'van gogh style painting of a yellow sunflower',
+        #         0.5,
+        #         0.5,
+        #         'van gogh',
+        #         10,
+        #         10,
+        #         7.5,
+        #     ],
+        # ]
         gr.Examples(examples=examples,
                     inputs=[
                         model_id,

example.py

@@ -0,0 +1,57 @@
+style_example = [
+            [
+                'CompVis/stable-diffusion-v1-4',
+                'FateZero/data/teaser_car-turn',
+                'a silver jeep driving down a curvy road in the countryside',
+                'watercolor painting of a silver jeep driving down a curvy road in the countryside',
+                0.8, 
+                0.8,
+                "watercolor",
+                10,
+                10,
+                7.5,
+                # input video argument
+                None, 0, 8, 1, 0,0,0,0
+                
+            ],
+            [
+                'CompVis/stable-diffusion-v1-4',
+                'FateZero/data/style/sunflower',
+                'a yellow sunflower',
+                'van gogh style painting of a yellow sunflower',
+                0.5,
+                0.5,
+                'van gogh',
+                10,
+                10,
+                7.5,
+                None, 0, 8, 1, 0,0,0,0
+            ],
+            [
+                'CompVis/stable-diffusion-v1-4',
+                'FateZero/data/style/surf',
+                'a man with round helmet surfing on a white wave in blue ocean with a rope',
+                'The Ukiyo-e style painting of a man with round helmet surfing on a white wave in blue ocean with a rope',
+                0.9,
+                0.9,
+                'Ukiyo-e',
+                10,
+                10,
+                7.5,
+                None, 0, 8, 1, 0,0,0,0
+            ],
+            [
+                'CompVis/stable-diffusion-v1-4',
+                'FateZero/FateZero/data/style/train',
+                'a train traveling down tracks next to a forest filled with trees and flowers and a man on the side of the track',
+                'a train traveling down tracks next to a forest filled with trees and flowers and a man on the side of the track Makoto Shinkai style',
+                0.9,
+                0.9,
+                'Makoto Shinkai',
+                10,
+                10,
+                7.5,
+                None, 0, 8, 1, 0,0,0,0
+            ],
+
+]