ghostsInTheMachine commited on
Commit
7bb1989
1 Parent(s): 17eeb1a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -10,6 +10,7 @@ from PIL import Image
10
  from concurrent.futures import ThreadPoolExecutor
11
  import moviepy.editor as mp
12
  from infer import lotus # Import the depth model inference function
 
13
 
14
  # Set device to use the L40s GPU
15
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
10
  from concurrent.futures import ThreadPoolExecutor
11
  import moviepy.editor as mp
12
  from infer import lotus # Import the depth model inference function
13
+ import spaces
14
 
15
  # Set device to use the L40s GPU
16
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")