Update app.py

app.py

@@ -203,6 +203,7 @@ def get_end_number(track_pause_number_slider, video_state, interactive_state):
 
     return video_state["painted_images"][track_pause_number_slider],interactive_state, operation_log, operation_log
 
+@spaces.GPU
 # use sam to get the mask
 def sam_refine(video_state, point_prompt, click_state, interactive_state, evt:gr.SelectData):
     """
@@ -265,6 +266,7 @@ def remove_multi_mask(interactive_state, mask_dropdown):
     operation_log = [("",""), ("Remove all masks. Try to add new masks","Normal")]
     return interactive_state, gr.update(choices=[],value=[]), operation_log, operation_log
 
+@spaces.GPU
 def show_mask(video_state, interactive_state, mask_dropdown):
     mask_dropdown.sort()
     select_frame = video_state["origin_images"][video_state["select_frame_number"]]
@@ -276,6 +278,7 @@ def show_mask(video_state, interactive_state, mask_dropdown):
     operation_log = [("",""), ("Added masks {}. If you want to do the inpainting with current masks, please go to step3, and click the Tracking button first and then Inpainting button.".format(mask_dropdown),"Normal")]
     return select_frame, operation_log, operation_log
 
+@spaces.GPU
 # tracking vos
 def vos_tracking_video(video_state, interactive_state, mask_dropdown):
     operation_log = [("",""), ("Tracking finished! Try to click the Inpainting button to get the inpainting result.","Normal")]
@@ -337,6 +340,7 @@ def vos_tracking_video(video_state, interactive_state, mask_dropdown):
     #### shanggao code for mask save
     return video_output, video_state, interactive_state, operation_log, operation_log
 
+@spaces.GPU
 def inpaint_video(video_state, *_):
     operation_log = [("", ""), ("Inpainting finished!", "Normal")]
 
@@ -397,6 +401,7 @@ def generate_video_from_frames(frames, output_path, fps=30):
     torchvision.io.write_video(output_path, frames, fps=fps, video_codec="libx264")
     return output_path
 
+@spaces.GPU
 def restart():
     operation_log = [("",""), ("Try to upload your video and click the Get video info button to get started!", "Normal")]
     return {