derekl35 HF Staff commited on
Commit
081238e
·
verified ·
1 Parent(s): 7804bc3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -9,6 +9,7 @@ import random
9
  from PIL import Image
10
  import os
11
  import time
 
12
 
13
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
14
  print(f"Using device: {DEVICE}")
 
9
  from PIL import Image
10
  import os
11
  import time
12
+ import spaces
13
 
14
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
15
  print(f"Using device: {DEVICE}")