hasnanmr commited on
Commit
c04578d
1 Parent(s): 444ea15

updating pickle load

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ transform = transforms.Compose([
44
  ])
45
 
46
  # Load the database of embeddings
47
- with open('face_database_ViT6.pkl', 'rb') as f:
48
  database = pickle.load(f)
49
 
50
  def cosine_similarity(embedding1, embedding2):
 
44
  ])
45
 
46
  # Load the database of embeddings
47
+ with open('face_database_me.pkl', 'rb') as f:
48
  database = pickle.load(f)
49
 
50
  def cosine_similarity(embedding1, embedding2):