ttoosi commited on
Commit
ee9d2e9
·
1 Parent(s): 8cd11f9

added vggcrop split=val

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ from datasets import load_dataset
12
  import random
13
 
14
  # Load the VGGFace2 dataset using Hugging Face's datasets library
15
- ds = load_dataset("chronopt-research/cropped-vggface2-224")
16
 
17
 
18
  # Load the model checkpoint from Hugging Face
 
12
  import random
13
 
14
  # Load the VGGFace2 dataset using Hugging Face's datasets library
15
+ ds = load_dataset("chronopt-research/cropped-vggface2-224", split="validation")
16
 
17
 
18
  # Load the model checkpoint from Hugging Face