ghostsInTheMachine
commited on
Commit
•
7bb1989
1
Parent(s):
17eeb1a
Update app.py
Browse files
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")
|