jhonparra18 commited on
Commit
c29054b
1 Parent(s): 4ab83eb

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +4 -1
inference.py CHANGED
@@ -5,13 +5,16 @@ import gdown
5
  import os
6
 
7
  cur_path=os.getcwd()
 
 
8
  ## download weights and config file
9
  file_id = '1GZusQVTXMTL8ydnCI4GkwCy6fyTNzEPC'
10
  url = f'https://drive.google.com/uc?id='
11
  output = f'{cur_path}/best_model.pth'
12
  gdown.download(f"{url}{file_id}", output, quiet=False)
13
 
14
- file_id = '1-Jot_x7bkWUE5SyAcKaaAAC-_T8DSEYe'
 
15
  url = f'https://drive.google.com/uc?id='
16
  output = f'{cur_path}/config.json'
17
  gdown.download(f"{url}{file_id}", output, quiet=False)
 
5
  import os
6
 
7
  cur_path=os.getcwd()
8
+
9
+
10
  ## download weights and config file
11
  file_id = '1GZusQVTXMTL8ydnCI4GkwCy6fyTNzEPC'
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 = '1-6-cjb0fqLba-KD8WyusRMB_AaIbIds2'
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)