change to decorator

app.py

@@ -26,6 +26,7 @@ import tempfile
 import cv2
 import matplotlib.pyplot as plt
 import numpy as np
+import spaces
 
 import torch
 
@@ -174,6 +175,7 @@ def clear_points(
     )
 
 
+@spaces.GPU(duration=10)
 def preprocess_video_in(
     video_path,
     first_frame,
@@ -258,6 +260,7 @@ def preprocess_video_in(
     ]
 
 
+@spaces.GPU(duration=5)
 def segment_with_points(
     point_type,
     first_frame,
@@ -350,6 +353,7 @@ def show_mask(mask, obj_id=None, random_color=False, convert_to_image=True):
     return mask
 
 
+@spaces.GPU(duration=30)
 def propagate_to_all(
     video_in,
     first_frame,
@@ -418,16 +422,6 @@ def propagate_to_all(
     )
 
 
-try:
-    from spaces import GPU
-
-    preprocess_video_in = GPU(preprocess_video_in, duration=10)
-    segment_with_points = GPU(segment_with_points, duration=5)
-    propagate_to_all = GPU(propagate_to_all, duration=30)
-except:
-    print("spaces unavailable")
-
-
 def update_ui():
     return gr.update(visible=True)