Add requirements
Browse files- app.py +1 -1
- 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=
|
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
|