osamaifti commited on
Commit
a6f78ee
·
1 Parent(s): f31e82d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ RECOGNITION_THRESHOLD = 0.3
12
 
13
 
14
  # Load the embedding model
15
- embedding_model = torch.load('final_modelnew', map_location=torch.device('cpu'))
16
  embedding_model.eval() # Set the model to evaluation mode
17
 
18
  # Database to store embeddings and user IDs
 
12
 
13
 
14
  # Load the embedding model
15
+ embedding_model = torch.load('final_modelnew.pth', map_location=torch.device('cpu'))
16
  embedding_model.eval() # Set the model to evaluation mode
17
 
18
  # Database to store embeddings and user IDs