update for zerogpu

app.py

@@ -43,6 +43,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))
+    infer_video_gpu = spaces.GPU(functools.partial(infer_video))
     gradio_theme = gr.themes.Default()
 
     with gr.Blocks(
@@ -143,7 +145,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"))
@@ -189,7 +191,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"))
@@ -201,7 +203,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],
         )
@@ -214,7 +216,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,