Update app.py

app.py

@@ -8,7 +8,7 @@ from typing import Iterable
 from tempfile import NamedTemporaryFile
 from PIL import Image
 from nudenet import NudeDetector
-from parallax import generate_parallax_images
+from parallax import generate_parallax_images, generate_animation_images
 
 
 class Theme(Base):
@@ -49,8 +49,8 @@ class Theme(Base):
         )
 
 
-@spaces.GPU(duration=30)
-def generate(image: np.ndarray, n: int):
+@spaces.GPU
+def generate_parallax(image: np.ndarray, n: int):
     input_image = Image.fromarray(image, 'RGBA')
     detector = NudeDetector()
     layers = []
@@ -75,18 +75,35 @@ def generate(image: np.ndarray, n: int):
     return [layers, json.dumps(indexes), nsfw]
 
 
+#@spaces.GPU
+def generate_animation(image: np.ndarray):
+    return generate_animation_images(Image.fromarray(image, 'RGBA'))
+
+
 with gr.Blocks(theme=Theme()) as demo:
+    gr.Markdown('**Parallax**')
+
+    with gr.Row():
+        with gr.Column():
+            parallax_input_image = gr.Image(image_mode='RGBA', label='Input')
+            parallax_input_layers = gr.Number(value=5, precision=0, minimum=2, maximum=10, step=1, label='Layers')
+            parallax_generate_button = gr.Button('Generate')
+        with gr.Column():
+            parallax_output_gallery = gr.Gallery(label='Outputs', columns=5)
+            parallax_output_json = gr.JSON(label='Indexes')
+            parallax_output_nsfw = gr.Checkbox(label='NSFW')
+
+    gr.Markdown('**Blink**')
+
     with gr.Row():
         with gr.Column():
-            input_image = gr.Image(image_mode='RGBA', label='Input')
-            input_layers = gr.Number(value=5, precision=0, minimum=2, maximum=10, step=1, label='Layers')
-            generate_button = gr.Button('Generate')
+            animation_input_image = gr.Image(image_mode='RGBA', label='Input')
+            animation_generate_button = gr.Button('Generate')
         with gr.Column():
-            output_gallery = gr.Gallery(label='Outputs', columns=5)
-            output_json = gr.JSON(label='Indexes')
-            output_nsfw = gr.Checkbox(label='NSFW')
+            animation_output_gallery = gr.Gallery(label='Outputs', columns=5)
 
-    generate_button.click(fn=generate, inputs=[input_image, input_layers], outputs=[output_gallery, output_json, output_nsfw], api_name='generate')
+    parallax_generate_button.click(fn=generate_parallax, inputs=[parallax_input_image, parallax_input_layers], outputs=[parallax_output_gallery, parallax_output_json, parallax_output_nsfw], api_name='generate_parallax')
+    animation_generate_button.click(fn=generate_animation, inputs=animation_input_image, outputs=animation_output_gallery, api_name='generate_animation')
 
 
 if __name__ == '__main__':