vchiang001 commited on
Commit
9f10359
·
1 Parent(s): bc2d71a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -542,7 +542,8 @@ def predict_pipeline(img_input,
542
  # Compute kpts for each crop
543
  list_kpts_per_crop = predict_dlc(list_crops,
544
  kpts_likelihood_th,
545
- path_to_DLCmodel,
 
546
  dlc_proc)
547
 
548
  img_background = img_input.resize((md_results.ims[0].shape[1],md_results.ims[0].shape[0]))
 
542
  # Compute kpts for each crop
543
  list_kpts_per_crop = predict_dlc(list_crops,
544
  kpts_likelihood_th,
545
+ "dlcmodel/models/DLC_ma_superquadruped_resnet_50_iteration-0_shuffle-1",
546
+ #path_to_DLCmodel,
547
  dlc_proc)
548
 
549
  img_background = img_input.resize((md_results.ims[0].shape[1],md_results.ims[0].shape[0]))