Spaces:
Runtime error
Runtime error
Commit
·
65af420
1
Parent(s):
9baf63e
Update app.py
Browse files
app.py
CHANGED
|
@@ -80,11 +80,11 @@ def download_file(session, file_spec, use_alt_url=False, chunk_size=128, num_att
|
|
| 80 |
pass
|
| 81 |
|
| 82 |
print('Downloading SceneDreamer pretrained model...')
|
| 83 |
-
|
| 84 |
-
|
| 85 |
-
|
| 86 |
-
|
| 87 |
-
|
| 88 |
|
| 89 |
|
| 90 |
|
|
|
|
| 80 |
pass
|
| 81 |
|
| 82 |
print('Downloading SceneDreamer pretrained model...')
|
| 83 |
+
with requests.Session() as session:
|
| 84 |
+
try:
|
| 85 |
+
download_file(session, pretrained_model)
|
| 86 |
+
except:
|
| 87 |
+
print('Google Drive download failed.\n')
|
| 88 |
|
| 89 |
|
| 90 |
|