Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ import tempfile
|
|
4 |
import torch
|
5 |
from torchvision import transforms
|
6 |
from mtcnn import MTCNN
|
7 |
-
from skimage import hog
|
8 |
import pickle
|
9 |
import torch.nn as nn
|
10 |
from torchvision.models import resnet50
|
@@ -79,9 +79,9 @@ def verify(img1, img2, model_type):
|
|
79 |
# st.write(f"Confidence: {prediction.item():.4f}")
|
80 |
|
81 |
if model_type == "HOG-SVM":
|
82 |
-
with open('
|
83 |
svm = pickle.load(f)
|
84 |
-
with open('
|
85 |
pca = pickle.load(f)
|
86 |
|
87 |
face1 = preprocess_image_svm(face1)
|
|
|
4 |
import torch
|
5 |
from torchvision import transforms
|
6 |
from mtcnn import MTCNN
|
7 |
+
from skimage.feature import hog
|
8 |
import pickle
|
9 |
import torch.nn as nn
|
10 |
from torchvision.models import resnet50
|
|
|
79 |
# st.write(f"Confidence: {prediction.item():.4f}")
|
80 |
|
81 |
if model_type == "HOG-SVM":
|
82 |
+
with open('svm.pkl', 'rb') as f:
|
83 |
svm = pickle.load(f)
|
84 |
+
with open('pca.pkl', 'rb') as f:
|
85 |
pca = pickle.load(f)
|
86 |
|
87 |
face1 = preprocess_image_svm(face1)
|