aharley commited on
Commit
933ba3a
·
1 Parent(s): 611054c

added gpu declaration

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -98,6 +98,7 @@ def get_colors(num_colors: int) -> List[Tuple[int, int, int]]:
98
  # )
99
  # return torch.stack([grid_x, grid_y], dim=-1).reshape(1, -1, 2)
100
 
 
101
  def paint_point_track_gpu_scatter(
102
  frames: np.ndarray,
103
  point_tracks: np.ndarray,
 
98
  # )
99
  # return torch.stack([grid_x, grid_y], dim=-1).reshape(1, -1, 2)
100
 
101
+ @spaces.GPU
102
  def paint_point_track_gpu_scatter(
103
  frames: np.ndarray,
104
  point_tracks: np.ndarray,