Update gradio_utils/utils.py

gradio_utils/utils.py

@@ -1,3 +1,4 @@
+import spaces
 import random
 import collections
 import gradio as gr
@@ -37,7 +38,7 @@ def adj_mx_from_edges(num_pts, skeleton, device='cuda', normalization_fix=True):
     return adj
 
 
-@spaces.GPU
+@spaces.GPU(duration=30)
 def plot_results(support_img, query_img, support_kp, support_w, query_kp, query_w,
                  skeleton=None, prediction=None, radius=6, in_color=None,
                  original_skeleton=None, img_alpha=0.6, target_keypoints=None):
@@ -127,7 +128,7 @@ def plot_results(support_img, query_img, support_kp, support_w, query_kp, query_
         plt.subplots_adjust(0, 0, 1, 1, 0, 0)
         return plt
 
-@spaces.GPU
+@spaces.GPU(duration=30)
 def process(query_img, state,
             cfg_path='configs/test/1shot_split1.py',
             checkpoint_path='ckpt/1shot_split1.pth'):