fixing image input

app.py

@@ -66,12 +66,15 @@ def compare_embeddings(embedding, database, threshold=0.9):
     return None, None
 
 def align_faces(frame, mtcnn, device):
+    # Convert the frame to a PIL image if it's a numpy array
+    if isinstance(frame, np.ndarray):
+        frame = Image.fromarray(frame)
     boxes, _ = mtcnn.detect(frame)
     aligned_faces = []
     if boxes is not None:
         aligned_faces = mtcnn(frame)
         if aligned_faces is not None:
-            aligned_faces = aligned_faces.to(device)
+            aligned_faces = [aligned_face.to(device) for aligned_face in aligned_faces]
     return aligned_faces, boxes
 
 def draw_annotations(frame, detections, names=None):
@@ -122,7 +125,7 @@ iface = gr.Interface(
     inputs=gr.Image(type="pil"),
     outputs=gr.Image(type="numpy"),
     title="Face Detection and Recognition with MTCNN and ViT",
-     description="Upload an image and the model will detect and align faces in it."
+    description="Upload an image and the model will detect and recognize faces in it."
 )
 
 # Launch the interface