reduce inference time

app.py

@@ -238,13 +238,15 @@ class GlobalState:
 
 
 @torch.no_grad()
-@spaces.GPU(duration=240)
+@spaces.GPU(duration=300)
 def process1(input_path, prompt, sd_model, seed, image_resolution, control_strength,
              x0_strength, control_type, low_threshold, high_threshold,
              ddpm_steps, scale, a_prompt, n_prompt,
              frame_count, batch_size, mininterv, maxinterv,
              use_constraints, bg_smooth, 
              b1, b2, s1, s2):
+    image_resolution = min(image_resolution, 512)
+    frame_count = min(frame_count, 8)
     global_state = GlobalState()
     global_state.update_controlnet_model(control_type)
     global_state.update_sd_model(sd_model)
@@ -569,12 +571,18 @@ with block:
                     b1, b2, s1, s2
                 ]
 
-                example = gr.Label('')
-
         with gr.Column():
             result_keyframe = gr.Video(label='Output key frame video',
                                        format='mp4',
                                        interactive=False)
+    with gr.Row():
+        example = gr.Examples(
+            examples=example_list,
+            inputs=[*ips],
+            fn=process1,
+            outputs=[result_keyframe],
+            cache_examples=True
+        )
 
     def input_changed(path):
         if path is None:
@@ -601,13 +609,7 @@ with block:
                                  value=frame_count,
                                  maximum=frame_count),
                 )
-    example = gr.Examples(
-        examples=example_list,
-        inputs=[*ips],
-        fn=process1,
-        outputs=[result_keyframe],
-        cache_examples=True
-    )
+    
 
     input_path.change(input_changed, input_path, [
                       mininterv, maxinterv, frame_count])

src/utils.py

@@ -40,7 +40,7 @@ def resize_image(input_image, resolution):
     H, W, C = input_image.shape
     H = float(H)
     W = float(W)
-    k = float(resolution) / min(H, W)
+    k = float(resolution) / max(H, W)
     H *= k
     W *= k
     H = int(np.round(H / 64.0)) * 64