jhonparra18 commited on
Commit
53563e5
1 Parent(s): b58bd7e

Update inference.py

Browse files
Files changed (1) hide show
  1. 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 547000
11
- file_id = '1Y4ss_MvyYuFtXdbcX_t9EpIs8O5A30qY'
 
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)