Spaces:
Running
Running
Minor change
Browse files
app.py
CHANGED
@@ -59,7 +59,7 @@ def get_image_features(model, image_dir):
|
|
59 |
|
60 |
loader = torch.utils.data.DataLoader(
|
61 |
dataset,
|
62 |
-
batch_size=
|
63 |
shuffle=False,
|
64 |
num_workers=8,
|
65 |
persistent_workers=True,
|
|
|
59 |
|
60 |
loader = torch.utils.data.DataLoader(
|
61 |
dataset,
|
62 |
+
batch_size=32,
|
63 |
shuffle=False,
|
64 |
num_workers=8,
|
65 |
persistent_workers=True,
|