Spaces:

nqtruong
/

Face_Image_Retrieval

Running

nqtruong commited on Jul 22, 2024

Commit

e5ecc25

verified ·

1 Parent(s): 22893bb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -35,7 +35,8 @@ class SiameseModel(nn.Module):
         return x
 model = SiameseModel()
-model.load_state_dict(torch.load('model_best_weights_1000.pt', map_location=torch.device('cpu')))
 model.eval()

         return x
 model = SiameseModel()
+model.load_state_dict(torch.load('model.pt', map_location=torch.device('cpu')))
+print("____________LOADED___________________")
 model.eval()