Update app.py
Browse files
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=
|
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 |
|