Spaces:

HaohuaLv
/

Scribble_Matching

Sleeping

HaohuaLv commited on Oct 24, 2023

Commit

a59047e

1 Parent(s): 9866292

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,7 @@ def load_candidates(candidate_dir, progress=gr.Progress()):
     dataset = Dataset.from_list(dataset)
     progress.tqdm(dataset)
     with torch.no_grad():
-        dataset = dataset.map(preprocess, batched=True, batch_size=1)
     return dataset

     dataset = Dataset.from_list(dataset)
     progress.tqdm(dataset)
     with torch.no_grad():
+        dataset = dataset.map(preprocess, batched=True, batch_size=64)
     return dataset