Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -16,6 +16,22 @@ def greet(n):
|
|
16 |
print(zero.device) # <-- 'cuda:0' 🤗
|
17 |
|
18 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
19 |
|
20 |
|
21 |
initialize_environment()
|
|
|
16 |
print(zero.device) # <-- 'cuda:0' 🤗
|
17 |
|
18 |
|
19 |
+
os.makedirs('/tmp/stf',exist_ok=True)
|
20 |
+
os.makedirs('/tmp/stf/TEMP',exist_ok=True)
|
21 |
+
|
22 |
+
file_id=os.getenv('DOWNLOAD_1')
|
23 |
+
url = f"https://drive.google.com/uc?id={file_id}"
|
24 |
+
gdown.download(url, '/tmp/stf-api-alternative.zip', quiet=False)
|
25 |
+
zipfile.ZipFile('/tmp/stf-api-alternative.zip', 'r').extractall('/tmp/stf/')
|
26 |
+
|
27 |
+
file_id=os.getenv('DOWNLOAD_2')
|
28 |
+
url = f"https://drive.google.com/uc?id={file_id}"
|
29 |
+
gdown.download(url, '/tmp/stf/089.pth', quiet=False)
|
30 |
+
|
31 |
+
|
32 |
+
file_id=os.getenv('DOWNLOAD_3')
|
33 |
+
url = f"https://drive.google.com/uc?id={file_id}"
|
34 |
+
gdown.download(url, '/tmp/stf/TEMP/0157.pth', quiet=False)
|
35 |
|
36 |
|
37 |
initialize_environment()
|