Add time measurement and exception handling to image_search_performers function

app.py

@@ -1,5 +1,6 @@
 import os
 import io
+import time
 import json
 import math
 import base64
@@ -50,11 +51,16 @@ def image_search_performers(image, threshold=20.0, results=3):
     image_array = np.array(image)
 
     response = []
-    import time
+    
     t = time.time()
-    faces = DeepFace.represent(img_path = image_array, detector_backend='retinaface', model_name='Facenet512', normalization="Facenet2018")
-    # faces = DeepFace.represent(img_path = image_array, detector_backend='yolov8', model_name='Facenet512', normalization="Facenet2018")
-    # faces = DeepFace.represent(img_path = image_array, detector_backend='mtcnn', model_name='Facenet512', normalization="Facenet2018")
+    try:
+        faces = DeepFace.represent(img_path = image_array, detector_backend='retinaface', model_name='Facenet512', normalization="Facenet2018")
+        # faces = DeepFace.represent(img_path = image_array, detector_backend='yolov8', model_name='Facenet512', normalization="Facenet2018")
+        # faces = DeepFace.represent(img_path = image_array, detector_backend='mtcnn', model_name='Facenet512', normalization="Facenet2018")
+    except ValueError as e:
+        print(e)
+        raise gr.Error("No faces found in the image")
+    
     print(f"Time to find faces: {time.time() - t}")
     for face in faces:
         embedding = face['embedding']