Update app.py

app.py

@@ -24,7 +24,7 @@ def load_candidates(candidate_dir):
     dataset = [dict(image=Image.open(tempfile.name).convert("RGB").resize((224, 224))) for tempfile in candidate_dir]
     dataset = Dataset.from_list(dataset)
     with torch.no_grad():
-        dataset = dataset.map(preprocess, batched=True, batch_size=1024)
+        dataset = dataset.map(preprocess, batched=True, batch_size=64)
 
     return dataset