Kaushik066 commited on
Commit
4f61d1e
·
verified ·
1 Parent(s): 80af7a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -63,7 +63,7 @@ class FaceEmbeddingModel(torch.nn.Module):
63
  return torch.nn.functional.normalize(x) # Normalize for cosine similarity
64
 
65
  # Load the model
66
- model_pretrained = torch.load(model_path, map_location=device)
67
 
68
  # Define the ML model - Evaluation function
69
  def prod_function(transformer_model, prod_dl, prod_data):
 
63
  return torch.nn.functional.normalize(x) # Normalize for cosine similarity
64
 
65
  # Load the model
66
+ model_pretrained = torch.load(model_path, map_location=device, weights_only=False)
67
 
68
  # Define the ML model - Evaluation function
69
  def prod_function(transformer_model, prod_dl, prod_data):