yerang commited on
Commit
3d0bf13
·
verified ·
1 Parent(s): ff68179

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -0
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()