Spaces:
Sleeping
Sleeping
jhonparra18
commited on
Commit
•
53563e5
1
Parent(s):
b58bd7e
Update inference.py
Browse files- inference.py +3 -2
inference.py
CHANGED
@@ -7,8 +7,9 @@ import os
|
|
7 |
cur_path=os.getcwd()
|
8 |
|
9 |
|
10 |
-
## download weights and config file ##checkpoint
|
11 |
-
|
|
|
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)
|
|
|
7 |
cur_path=os.getcwd()
|
8 |
|
9 |
|
10 |
+
## download weights and config file ##checkpoint 565000
|
11 |
+
|
12 |
+
file_id = '1HqnOpMnRwqu93-VP1qqi3Kz6oyUZ2Rsq'
|
13 |
url = f'https://drive.google.com/uc?id='
|
14 |
output = f'{cur_path}/best_model.pth'
|
15 |
gdown.download(f"{url}{file_id}", output, quiet=False)
|