Spaces:
Sleeping
Sleeping
jhonparra18
commited on
Commit
•
9e660e7
1
Parent(s):
547bc75
new model weights
Browse files- inference.py +2 -2
inference.py
CHANGED
@@ -8,13 +8,13 @@ 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)
|
15 |
|
16 |
|
17 |
-
file_id = '
|
18 |
url = f'https://drive.google.com/uc?id='
|
19 |
output = f'{cur_path}/config.json'
|
20 |
gdown.download(f"{url}{file_id}", output, quiet=False)
|
|
|
8 |
|
9 |
|
10 |
## download weights and config file ##checkpoint 565000
|
11 |
+
file_id = '1ZuCX7bMegfem3_hGb6OT64gzpmNwRwrn'
|
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)
|
15 |
|
16 |
|
17 |
+
file_id = '1TMzqXDgqLwucusiG8HfAHBf5CozANAoe'
|
18 |
url = f'https://drive.google.com/uc?id='
|
19 |
output = f'{cur_path}/config.json'
|
20 |
gdown.download(f"{url}{file_id}", output, quiet=False)
|