Update inference.py

inference.py

@@ -7,14 +7,13 @@ import os
 cur_path=os.getcwd()
 
 
-## download weights and config file
-file_id = '1-uVNiBjN9d6GBZy3jubvgq7slyEgObyr'
+## download weights and config file ##checkpoint 547000
+file_id = '1Y4ss_MvyYuFtXdbcX_t9EpIs8O5A30qY'
 url = f'https://drive.google.com/uc?id='
 output = f'{cur_path}/best_model.pth'
 gdown.download(f"{url}{file_id}", output, quiet=False)
 
-
-file_id = '1-6-cjb0fqLba-KD8WyusRMB_AaIbIds2'
+file_id = '1OyOe5-f1nrJlU8dG8ZaIUz61zLGtGRTX'
 url = f'https://drive.google.com/uc?id='
 output = f'{cur_path}/config.json'
 gdown.download(f"{url}{file_id}", output, quiet=False)