Update app.py

app.py

@@ -1,11 +1,11 @@
 # Transformers and its models
-import transformers
+#import transformers
 
 # For Image Processing
-from transformers import ViTImageProcessor
+#from transformers import ViTImageProcessor
 
 # For Model
-from transformers import ViTModel, ViTConfig, pipeline
+#from transformers import ViTModel, ViTConfig, pipeline
 import insightface
 from insightface.app import FaceAnalysis
 
@@ -13,9 +13,9 @@ from insightface.app import FaceAnalysis
 from torchvision import transforms, datasets
 
 # For GPU
-from transformers import set_seed
-from torch.optim import AdamW
-from accelerate import Accelerator, notebook_launcher
+#from transformers import set_seed
+#from torch.optim import AdamW
+#from accelerate import Accelerator, notebook_launcher
 
 # For Data Loaders
 import datasets
@@ -216,7 +216,7 @@ app.prepare(ctx_id=-1, det_size=(IMAGE_SHAPE, IMAGE_SHAPE))  # Use ctx_id=-1 if
 
 # Define the ML model - Evaluation function
 def prod_function(app, prod_path, webcam_path):
-    webcam_img = cv2.imread(webcam_path)
+    webcam_img = Image.open(webcam_path)
     webcam_emb = app.get(webcam_img, max_num=1)
     webcam_emb = webcam_emb[0].embedding