Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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):
|