Update app.py

app.py

@@ -187,6 +187,8 @@ def process_video_with_points(video_path, click_points, tracker):
     return output_path
 
 # -------------------- Model Initialization -------------------- #
+@torch.no_grad()
+@spaces.GPU
 def load_model():
     """Initialize and load the tracking model."""
     # Adjust these paths as needed.
@@ -215,6 +217,8 @@ def load_model():
     
     return model
 
+@torch.no_grad()
+@spaces.GPU
 def create_tracker(model):
     """Create tracker instance with the loaded model."""
     return Tracker(
@@ -251,6 +255,8 @@ def process_with_config(video_path, click_points, resolution, window_index, trac
     tracker.S = mapping.get(int(window_index), 16)
     return process_video_with_points(video_path, click_points, tracker)
 
+@torch.no_grad()
+@spaces.GPU
 def main():
     """Main function that initializes the model and runs the Gradio interface."""
     # Set torch matmul precision