cdnuts commited on
Commit
fa4c226
1 Parent(s): a909331

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -243,7 +243,7 @@ def process_zip(zip_file, threshold):
243
  def process_images_light(images, threshold):
244
  dataset = ImageDataset(images, transform)
245
 
246
- dataloader = DataLoader(dataset, batch_size=16, num_workers=0, pin_memory=True, drop_last=False)
247
 
248
  all_results = []
249
 
 
243
  def process_images_light(images, threshold):
244
  dataset = ImageDataset(images, transform)
245
 
246
+ dataloader = DataLoader(dataset, batch_size=32, num_workers=0, pin_memory=True, drop_last=False)
247
 
248
  all_results = []
249