Spaces:
No application file
No application file
Add employee_images folder with Git LFS
Browse files- app.py +3 -3
- requirements.txt +2 -1
app.py
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
|
2 |
import os # Import the os module
|
3 |
import torch
|
4 |
from facenet_pytorch import InceptionResnetV1, MTCNN
|
@@ -43,7 +43,7 @@ def register_images_facenet(model, image_dir):
|
|
43 |
return embeddings
|
44 |
|
45 |
# Load employee images and register them
|
46 |
-
employee_images_dir = '
|
47 |
facenet_embeddings = register_images_facenet(facenet_model, employee_images_dir)
|
48 |
|
49 |
# Identify image function
|
@@ -70,4 +70,4 @@ def identify_image_facenet(model, embeddings, img, threshold=0.5):
|
|
70 |
facenet_interface = gr.Interface(fn=identify_image_facenet, inputs="image", outputs="text", title="FaceNet Verification with Threshold 0.5")
|
71 |
|
72 |
if _name_ == "_main_":
|
73 |
-
facenet_interface.launch()
|
|
|
1 |
+
|
2 |
import os # Import the os module
|
3 |
import torch
|
4 |
from facenet_pytorch import InceptionResnetV1, MTCNN
|
|
|
43 |
return embeddings
|
44 |
|
45 |
# Load employee images and register them
|
46 |
+
employee_images_dir = '/content/employees_images' # Ensure this directory is in the same directory as app.py
|
47 |
facenet_embeddings = register_images_facenet(facenet_model, employee_images_dir)
|
48 |
|
49 |
# Identify image function
|
|
|
70 |
facenet_interface = gr.Interface(fn=identify_image_facenet, inputs="image", outputs="text", title="FaceNet Verification with Threshold 0.5")
|
71 |
|
72 |
if _name_ == "_main_":
|
73 |
+
facenet_interface.launch()
|
requirements.txt
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
facenet-pytorch
|
2 |
deepface
|
3 |
gradio
|
@@ -5,4 +6,4 @@ mtcnn
|
|
5 |
torch
|
6 |
torchvision
|
7 |
scikit-learn
|
8 |
-
pillow
|
|
|
1 |
+
|
2 |
facenet-pytorch
|
3 |
deepface
|
4 |
gradio
|
|
|
6 |
torch
|
7 |
torchvision
|
8 |
scikit-learn
|
9 |
+
pillow
|