update for zerogpu

app.py

@@ -42,6 +42,8 @@ def infer_video(path_input, seed):
     return [g_save_path, d_save_path]
 
 def run_demo_server():
+    infer_gpu = spaces.GPU(functools.partial(infer, seed=0))
+    infer_video_gpu = spaces.GPU(functools.partial(infer_video, seed=0))
     gradio_theme = gr.themes.Default()
 
     with gr.Blocks(
@@ -142,7 +144,7 @@ def run_demo_server():
                             )
 
                 gr.Examples(
-                    fn=infer,
+                    fn=infer_gpu,
                     examples=sorted([
                         [os.path.join("files", "images", name), 0]
                         for name in os.listdir(os.path.join("files", "images"))
@@ -188,7 +190,7 @@ def run_demo_server():
                             )
 
                 gr.Examples(
-                    fn=infer_video,
+                    fn=infer_video_gpu,
                     examples=sorted([
                         [os.path.join("files", "videos", name), 0]
                         for name in os.listdir(os.path.join("files", "videos"))
@@ -200,7 +202,7 @@ def run_demo_server():
 
         ### Image
         image_submit_btn.click(
-            fn=infer,
+            fn=infer_gpu,
             inputs=[image_input, seed],
             outputs=[image_output_g, image_output_d],
             concurrency_limit=1,
@@ -218,7 +220,7 @@ def run_demo_server():
 
         ### Video
         video_submit_btn.click(
-            fn=infer_video,
+            fn=infer_video_gpu,
             inputs=[input_video, seed],
             outputs=[video_output_g, video_output_d],
             queue=True,