add spaces

app.py

@@ -130,7 +130,9 @@ def infer(*user_args):
         input_traj = input_traj,
     )
 
+    print('GPU starts')
     video = infer_gpu_part(args)
+    print('GPU ends')
     
     video = torch.clamp(video.float(), -1.0, 1.0)
     video = video.permute(2, 0, 1, 3, 4)  # t,n,c,h,w
@@ -206,7 +208,7 @@ def infer(*user_args):
     return video_path, video_bbox_path
 
     
-@spaces.GPU(duration=270)
+@spaces.GPU(duration=250)
 def infer_gpu_part(args):
 
     model = instantiate_from_config(args.model_config_512)
@@ -622,7 +624,7 @@ with gr.Blocks(css=css) as demo:
                 
             with gr.Row(visible=False) as row_diy:
                 dropdown_diy = gr.Dropdown(
-                    label="Number of keyframes",
+                    label="Number of Keyframes",
                     choices=range(2, MAX_KEYS+1),
                 )
                 
@@ -699,7 +701,7 @@ with gr.Blocks(css=css) as demo:
                     with gr.Row():
                         ddim_steps = gr.Slider(label='DDIM Steps',
                                 minimum=5,
-                                maximum=200,
+                                maximum=50,
                                 step=1,
                                 value=50)
                         unconditional_guidance_scale = gr.Slider(label='Unconditional Guidance Scale',
@@ -723,7 +725,7 @@ with gr.Blocks(css=css) as demo:
             submit_btn = gr.Button("Generate", variant='primary')
 
         with gr.Row():
-            check_btn = gr.Button("Check Existing Results", variant='secondary')
+            check_btn = gr.Button("Check Existing Results (in case of the connection lost)", variant='secondary')
 
         with gr.Row():
             gr.Examples(label='Sample Prompts', examples=examples, inputs=[prompt_in, target_indices, ddim_edit, seed, ddim_steps, unconditional_guidance_scale, video_fps, save_fps, height_ratio, width_ratio, radio_mode, dropdown_diy, *frame_indices, *h_positions, *w_positions])

lvdm/models/samplers/ddim_freetraj.py

@@ -215,7 +215,6 @@ class DDIMSampler(object):
                       temperature=1., noise_dropout=0., score_corrector=None, corrector_kwargs=None,
                       unconditional_guidance_scale=1., unconditional_conditioning=None,
                       uc_type=None, conditional_guidance_scale_temporal=None, step=0, ddim_edit=0, **kwargs):
-        print(step)
         b, *_, device = *x.shape, x.device
         if x.dim() == 5:
             is_video = True