sgugger commited on
Commit
22d5448
1 Parent(s): b3195da

Add requirements

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +5 -0
app.py CHANGED
@@ -60,7 +60,7 @@ def process_dataset(dataset_name, dataset_split, dataset_column_image):
60
  return example
61
 
62
  # Compute hash of every image in the dataset
63
- img_dataset = img_dataset.map(add_hash, num_proc=4)
64
  return img_dataset
65
 
66
 
 
60
  return example
61
 
62
  # Compute hash of every image in the dataset
63
+ img_dataset = img_dataset.map(add_hash, num_proc=max(cpu_count() // 2, 1))
64
  return img_dataset
65
 
66
 
requirements.txt ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ gradio
2
+ scipy
3
+ numpy
4
+ datasets
5
+ Pillow