Stable-X commited on
Commit
6f6423c
·
verified ·
1 Parent(s): d9c3e1a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -15,6 +15,7 @@ from scipy.spatial.transform import Rotation
15
  from transformers import AutoModelForImageSegmentation
16
  from torchvision import transforms
17
  from PIL import Image
 
18
 
19
  # Default values
20
  DEFAULT_CKPT_PATH = './checkpoints/spann3r.pth'
 
15
  from transformers import AutoModelForImageSegmentation
16
  from torchvision import transforms
17
  from PIL import Image
18
+ import spaces
19
 
20
  # Default values
21
  DEFAULT_CKPT_PATH = './checkpoints/spann3r.pth'