Update app.py

app.py

@@ -36,7 +36,7 @@ def search(query: str, ds, images, k):
         
     qs = []
     with torch.no_grad():
-        batch_query = process_queries(processor, [query], mock_image)
+        batch_query = processor.process_queries([query])
         batch_query = {k: v.to(device) for k, v in batch_query.items()}
         embeddings_query = model(**batch_query)
         qs.extend(list(torch.unbind(embeddings_query.to("cpu"))))
@@ -80,7 +80,7 @@ def index_gpu(images, ds):
         images,
         batch_size=4,
         shuffle=False,
-        collate_fn=lambda x: process_images(processor, x),
+        collate_fn=processor.process_images,
     )