Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -82,7 +82,7 @@ def index_gpu(images, ds):
|
|
| 82 |
# run inference - docs
|
| 83 |
dataloader = DataLoader(
|
| 84 |
images,
|
| 85 |
-
batch_size=
|
| 86 |
shuffle=False,
|
| 87 |
collate_fn=lambda x: processor.process_images(x).to(model.device),
|
| 88 |
)
|
|
|
|
| 82 |
# run inference - docs
|
| 83 |
dataloader = DataLoader(
|
| 84 |
images,
|
| 85 |
+
batch_size=4,
|
| 86 |
shuffle=False,
|
| 87 |
collate_fn=lambda x: processor.process_images(x).to(model.device),
|
| 88 |
)
|