jhonparra18 commited on
Commit
9e660e7
1 Parent(s): 547bc75

new model weights

Browse files
Files changed (1) hide show
  1. 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 = '1jFhV3p9O6YZhTmK1DlJ5jq-Z2rqz-Av8'
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 = '1cnk0kzVkZVRl9ZWj1lHcnE2EmS28_0de'
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)