Update app.py

to avoid loading train

app.py

@@ -7,12 +7,18 @@ from huggingface_hub import hf_hub_download
 
 from PIL import Image
 import numpy as np
+import random
 
 from datasets import load_dataset
-import random
+from datasets import DatasetDict
+ds = DatasetDict({
+    "validation": load_dataset("chronopt-research/cropped-vggface2-224", split="validation"),
+})
+
+
 
 # Load the VGGFace2 dataset using Hugging Face's datasets library
-ds = load_dataset("chronopt-research/cropped-vggface2-224", split="validation")
+# ds = load_dataset("chronopt-research/cropped-vggface2-224", split="validation")
 
 
 # Load the model checkpoint from Hugging Face