Spaces:
Sleeping
Sleeping
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)
|