manu commited on
Commit
82bbd91
·
verified ·
1 Parent(s): a129662

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -141,6 +141,7 @@ def index_gpu(images, ds):
141
  dataloader = DataLoader(
142
  images,
143
  batch_size=4,
 
144
  shuffle=False,
145
  collate_fn=lambda x: processor.process_images(x).to(model.device),
146
  )
 
141
  dataloader = DataLoader(
142
  images,
143
  batch_size=4,
144
+ num_workers=4,
145
  shuffle=False,
146
  collate_fn=lambda x: processor.process_images(x).to(model.device),
147
  )