add spaces

app.py

@@ -47,6 +47,15 @@ def check_move(trajectory, video_length=16):
 
     return True
 
+def check(radio_mode):
+    if radio_mode == 'ori':
+        video_path = "output.mp4"
+        video_bbox_path = "output.mp4"
+    else:
+        video_path = "output_freetraj.mp4"
+        video_bbox_path = "output_freetraj_bbox.mp4"
+    return video_path, video_bbox_path
+    
 @spaces.GPU(duration=280)
 def infer(*user_args):
     prompt_in = user_args[0]
@@ -698,6 +707,7 @@ with gr.Blocks(css=css) as demo:
                 
         with gr.Row():
             submit_btn = gr.Button("Generate", variant='primary')
+            check_btn = gr.Button("Check Existing Results", variant='primary')
 
         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])
@@ -731,6 +741,11 @@ with gr.Blocks(css=css) as demo:
     submit_btn.click(fn=infer,
             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],
             outputs=[video_result, video_result_bbox],
-            api_name="freetraj")
+            api_name="generate")
+    
+    check_btn.click(fn=check,
+            inputs=[radio_mode],
+            outputs=[video_result, video_result_bbox],
+            api_name="check")
 
 demo.queue(max_size=8).launch(show_api=True)