osamaifti commited on
Commit
a5b31f5
1 Parent(s): 1a364b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ import os
10
 
11
  RECOGNITION_THRESHOLD = 0.3
12
 
13
- # Assuming the PyTorch model is a ResNet (or similar) and has been trained accordingly
14
  # Load the embedding model
15
  embedding_model = torch.load('full_mode2.pth', map_location=torch.device('cpu'))
16
  embedding_model.eval() # Set the model to evaluation mode
 
10
 
11
  RECOGNITION_THRESHOLD = 0.3
12
 
13
+
14
  # Load the embedding model
15
  embedding_model = torch.load('full_mode2.pth', map_location=torch.device('cpu'))
16
  embedding_model.eval() # Set the model to evaluation mode