Update app.py

app.py

@@ -4,6 +4,9 @@ import torch
 from torchvision import transforms, utils
 from facenet_pytorch import MTCNN
 from torchvision.transforms.functional import to_pil_image
+import cv2
+import numpy as np
+import io
 
 # Function to load the ViT model and MTCNN
 def load_model(model_path):