diff --git a/scface_test.py b/scface_test.py
index 7d6d55d..f485e38 100644
--- a/scface_test.py
+++ b/scface_test.py
@@ -170,7 +170,6 @@ try:
         mugshot_data, surveillance_data = pickle.load(file)
 except FileNotFoundError:
     for person in mugshot_data.keys():
-        print(f'##### Person {person} #####')
         img = preprocess_image(mugshot_data[person]['file'])
 
         embeddings_vit = vit_model(img).numpy()
@@ -187,5 +186,24 @@ except FileNotFoundError:
         mugshot_data[person]['embeddings']['mobilenet'] = embeddings1_mobilenet
         mugshot_data[person]['embeddings']['efficientnet'] = embeddings1_efficientnet
 
-    # for person in surveillance_data.keys():
-    #     pass
+    for person in surveillance_data.keys():
+        for file in surveillance_data[person].keys():
+            img = preprocess_image(surveillance_data[person][file]['file'])
+
+            embeddings_vit = vit_model(img).numpy()
+            embeddings1_resnet = resnet50_model(img).numpy()
+            embeddings1_vgg16 = vgg16_model(img).numpy()
+            embeddings1_inception = inception_model(img).numpy()
+            embeddings1_mobilenet = mobilenet_model(img).numpy()
+            embeddings1_efficientnet = efficientnetB0_model(img).numpy()
+
+            surveillance_data[person][file]['embeddings']['vit'] = embeddings_vit
+            surveillance_data[person][file]['embeddings']['resnet'] = embeddings1_resnet
+            surveillance_data[person][file]['embeddings']['vgg'] = embeddings1_vgg16
+            surveillance_data[person][file]['embeddings']['inception'] = embeddings1_inception
+            surveillance_data[person][file]['embeddings']['mobilenet'] = embeddings1_mobilenet
+            surveillance_data[person][file]['embeddings']['efficientnet'] = embeddings1_efficientnet
+
+    with open('./saved_results/Tests/SCface/embeddings.pickle', 'wb') as file:
+        data = (mugshot_data, surveillance_data)
+        pickle.dump(data, file)