Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -12,16 +12,16 @@ from download import download_files_from_url
|
|
12 |
zero = torch.Tensor([0]).cuda()
|
13 |
print(zero.device) # <-- 'cpu' π€
|
14 |
|
|
|
|
|
|
|
|
|
|
|
15 |
@spaces.GPU
|
16 |
def greet(n):
|
17 |
print(zero.device) # <-- 'cuda:0' π€
|
18 |
|
19 |
|
20 |
-
|
21 |
-
download_files_from_url()
|
22 |
-
|
23 |
-
initialize_environment()
|
24 |
-
|
25 |
sys.path.append('/home/user/.local/lib/python3.10/site-packages')
|
26 |
sys.path.append('/home/user/.local/lib/python3.10/site-packages/stf_alternative/src/stf_alternative')
|
27 |
sys.path.append('/home/user/.local/lib/python3.10/site-packages/stf_tools/src/stf_tools')
|
|
|
12 |
zero = torch.Tensor([0]).cuda()
|
13 |
print(zero.device) # <-- 'cpu' π€
|
14 |
|
15 |
+
download_files_from_url()
|
16 |
+
|
17 |
+
initialize_environment()
|
18 |
+
|
19 |
+
|
20 |
@spaces.GPU
|
21 |
def greet(n):
|
22 |
print(zero.device) # <-- 'cuda:0' π€
|
23 |
|
24 |
|
|
|
|
|
|
|
|
|
|
|
25 |
sys.path.append('/home/user/.local/lib/python3.10/site-packages')
|
26 |
sys.path.append('/home/user/.local/lib/python3.10/site-packages/stf_alternative/src/stf_alternative')
|
27 |
sys.path.append('/home/user/.local/lib/python3.10/site-packages/stf_tools/src/stf_tools')
|