nikkar commited on
Commit
542ca08
1 Parent(s): b3bcbf2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,7 +15,7 @@ import torch
15
  import colorsys
16
  import random
17
  from typing import List, Optional, Sequence, Tuple
18
-
19
  import numpy as np
20
 
21
 
@@ -318,7 +318,7 @@ def preprocess_video_input(video_path):
318
  gr.update(interactive=True),
319
  )
320
 
321
-
322
  def track(
323
  video_preview,
324
  video_input,
 
15
  import colorsys
16
  import random
17
  from typing import List, Optional, Sequence, Tuple
18
+ import spaces
19
  import numpy as np
20
 
21
 
 
318
  gr.update(interactive=True),
319
  )
320
 
321
+ @spaces.GPU
322
  def track(
323
  video_preview,
324
  video_input,