dvieri commited on
Commit
8aa0ccd
·
verified ·
1 Parent(s): c54f4ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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('svm_model.pkl', 'rb') as f:
83
  svm = pickle.load(f)
84
- with open('pca_model.pkl', 'rb') as f:
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)