Janne Mäyrä commited on
Commit
2f371ff
·
unverified ·
1 Parent(s): ba2ca67

fix amodel paths

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,12 +17,12 @@ def run_models(
17
  hp_result = hp_model(im[:,:,::-1], conf=conf_thr)
18
  hp_im = hp_result[0].plot()
19
 
20
- spk_model = YOLO(f'{model_path}{model_type}_spk/best.pt')
21
  spk_model.to(device='cpu')
22
  spk_result = spk_model(im[:,:,::-1], conf=conf_thr)
23
  spk_im = spk_result[0].plot()
24
 
25
- both_model = YOLO(f'{model_path}{model_type}_both/best.pt')
26
  both_model.to(device='cpu')
27
  both_result = both_model(im[:,:,::-1], conf=conf_thr)
28
  both_im = both_result[0].plot()
 
17
  hp_result = hp_model(im[:,:,::-1], conf=conf_thr)
18
  hp_im = hp_result[0].plot()
19
 
20
+ spk_model = YOLO(f'{model_path}{model_type}_spk.pt')
21
  spk_model.to(device='cpu')
22
  spk_result = spk_model(im[:,:,::-1], conf=conf_thr)
23
  spk_im = spk_result[0].plot()
24
 
25
+ both_model = YOLO(f'{model_path}{model_type}_both.pt')
26
  both_model.to(device='cpu')
27
  both_result = both_model(im[:,:,::-1], conf=conf_thr)
28
  both_im = both_result[0].plot()