Spaces:

manu
/

vision-rag-docker-image

Sleeping

manu commited on Feb 25

Commit

c845a99

verified ·

1 Parent(s): c531d50

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -133,7 +133,7 @@ def index_gpu(images, ds):
     # run inference - docs
     dataloader = DataLoader(
         images,
-        batch_size=16,
         shuffle=False,
         collate_fn=lambda x: processor.process_images(x).to(model.device),
     )

     # run inference - docs
     dataloader = DataLoader(
         images,
+        batch_size=4,
         shuffle=False,
         collate_fn=lambda x: processor.process_images(x).to(model.device),
     )