limit param

app.py

@@ -98,7 +98,7 @@ vis = FaceMeshVisualizer()
 
 frame_inter_model = init_frame_interpolation_model()
 
-@spaces.GPU(duration=300)
+@spaces.GPU(duration=100)
 def audio2video(input_audio, ref_img, headpose_video=None, size=512, steps=25, length=60, seed=42):
     fps = 30
     cfg = 3.5
@@ -164,7 +164,7 @@ def audio2video(input_audio, ref_img, headpose_video=None, size=512, steps=25, l
     #     [transforms.Resize((height, width)), transforms.ToTensor()]
     # )
     args_L = len(pose_images) if length==0 or length > len(pose_images) else length
-    args_L = min(args_L, 90)
+    args_L = min(args_L, 60)
     for pose_image_np in pose_images[: args_L : fi_step]:
         # pose_image_pil = Image.fromarray(cv2.cvtColor(pose_image_np, cv2.COLOR_BGR2RGB))
         # pose_tensor_list.append(pose_transform(pose_image_pil))
@@ -209,7 +209,7 @@ def audio2video(input_audio, ref_img, headpose_video=None, size=512, steps=25, l
     
     return save_path.replace('_noaudio.mp4', '.mp4'), ref_image_pil
 
-@spaces.GPU(duration=300)
+@spaces.GPU(duration=100)
 def video2video(ref_img, source_video, size=512, steps=25, length=60, seed=42):
     cfg = 3.5
     fi_step = 3
@@ -257,7 +257,7 @@ def video2video(ref_img, source_video, size=512, steps=25, length=60, seed=42):
     verts_list = []
     bs_list = []
     args_L = len(source_images) if length==0 or length*step > len(source_images) else length*step
-    args_L = min(args_L, 90*step)
+    args_L = min(args_L, 60*step)
     for src_image_pil in source_images[: args_L : step*fi_step]:
         src_img_np = cv2.cvtColor(np.array(src_image_pil), cv2.COLOR_RGB2BGR)
         frame_height, frame_width, _ = src_img_np.shape
@@ -379,11 +379,11 @@ with gr.Blocks() as demo:
                     a2v_headpose_video = gr.Video(label="Option: upload head pose reference video", sources="upload")
 
                 with gr.Row():
-                    a2v_size_slider = gr.Slider(minimum=256, maximum=1024, step=8, value=512, label="Video size (-W & -H)")
+                    a2v_size_slider = gr.Slider(minimum=256, maximum=512, step=8, value=512, label="Video size (-W & -H)")
                     a2v_step_slider = gr.Slider(minimum=5, maximum=20, step=1, value=15, label="Steps (--steps)")
                 
                 with gr.Row():
-                    a2v_length = gr.Slider(minimum=0, maximum=90, step=1, value=30, label="Length (-L)")
+                    a2v_length = gr.Slider(minimum=0, maximum=60, step=1, value=30, label="Length (-L)")
                     a2v_seed = gr.Number(value=42, label="Seed (--seed)")
                 
                 a2v_botton = gr.Button("Generate", variant="primary")
@@ -407,11 +407,11 @@ with gr.Blocks() as demo:
                     v2v_source_video = gr.Video(label="Upload source video", sources="upload")
                 
                 with gr.Row():
-                    v2v_size_slider = gr.Slider(minimum=256, maximum=1024, step=8, value=512, label="Video size (-W & -H)")
+                    v2v_size_slider = gr.Slider(minimum=256, maximum=512, step=8, value=512, label="Video size (-W & -H)")
                     v2v_step_slider = gr.Slider(minimum=5, maximum=20, step=1, value=15, label="Steps (--steps)")
                 
                 with gr.Row():
-                    v2v_length = gr.Slider(minimum=0, maximum=90, step=1, value=30, label="Length (-L)")
+                    v2v_length = gr.Slider(minimum=0, maximum=60, step=1, value=30, label="Length (-L)")
                     v2v_seed = gr.Number(value=42, label="Seed (--seed)")
                 
                 v2v_botton = gr.Button("Generate", variant="primary")