jayabbhi commited on
Commit
c3a6003
·
verified ·
1 Parent(s): 75e546c

restoring to defualt

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -77,7 +77,7 @@ def infer(in_pil_img, in_threshold=0.8, out_pil_img=None):
77
 
78
  # Run HMR2.0 on all detected humans
79
  dataset = ViTDetDataset(model_cfg, open_cv_image, boxes)
80
- dataloader = torch.utils.data.DataLoader(dataset, batch_size=1, shuffle=False, num_workers=0)
81
 
82
  all_verts = []
83
  all_cam_t = []
 
77
 
78
  # Run HMR2.0 on all detected humans
79
  dataset = ViTDetDataset(model_cfg, open_cv_image, boxes)
80
+ dataloader = torch.utils.data.DataLoader(dataset, batch_size=8, shuffle=False, num_workers=0)
81
 
82
  all_verts = []
83
  all_cam_t = []