Spaces:
Sleeping
Sleeping
Update inference.py
Browse files- inference.py +1 -2
inference.py
CHANGED
|
@@ -6,9 +6,8 @@ import os
|
|
| 6 |
|
| 7 |
cur_path=os.getcwd()
|
| 8 |
|
| 9 |
-
|
| 10 |
## download weights and config file ##checkpoint 565000
|
| 11 |
-
file_id = '
|
| 12 |
url = f'https://drive.google.com/uc?id='
|
| 13 |
output = f'{cur_path}/best_model.pth'
|
| 14 |
gdown.download(f"{url}{file_id}", output, quiet=False)
|
|
|
|
| 6 |
|
| 7 |
cur_path=os.getcwd()
|
| 8 |
|
|
|
|
| 9 |
## download weights and config file ##checkpoint 565000
|
| 10 |
+
file_id = '1HqnOpMnRwqu93-VP1qqi3Kz6oyUZ2Rsq'
|
| 11 |
url = f'https://drive.google.com/uc?id='
|
| 12 |
output = f'{cur_path}/best_model.pth'
|
| 13 |
gdown.download(f"{url}{file_id}", output, quiet=False)
|