Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -71,7 +71,7 @@ def index(file, ds):
|
|
| 71 |
# run inference - docs
|
| 72 |
dataloader = DataLoader(
|
| 73 |
images,
|
| 74 |
-
batch_size=
|
| 75 |
shuffle=False,
|
| 76 |
collate_fn=lambda x: process_images(processor, x),
|
| 77 |
)
|
|
|
|
| 71 |
# run inference - docs
|
| 72 |
dataloader = DataLoader(
|
| 73 |
images,
|
| 74 |
+
batch_size=8,
|
| 75 |
shuffle=False,
|
| 76 |
collate_fn=lambda x: process_images(processor, x),
|
| 77 |
)
|