manu commited on
Commit
ba29302
·
verified ·
1 Parent(s): 8666c3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -194,6 +194,7 @@ def index_gpu(images, ds):
194
  dataloader = DataLoader(
195
  images,
196
  batch_size=4,
 
197
  shuffle=False,
198
  collate_fn=lambda x: processor.process_images(x).to(model.device),
199
  )
 
194
  dataloader = DataLoader(
195
  images,
196
  batch_size=4,
197
+ num_workers=4,
198
  shuffle=False,
199
  collate_fn=lambda x: processor.process_images(x).to(model.device),
200
  )