Update main.py

main.py

@@ -22,18 +22,18 @@ with open('list_eval_partition.txt', 'r') as f:
 # For a model pretrained on VGGFace2
 print('Loading model weights ........')
 
-class SiameseModel(nn.Module):
-    def __init__(self):
-        super().__init__()
-        self.backbone = InceptionResnetV1(pretrained='vggface2')
-    def forward(self, x):
-        x = self.backbone(x)
-        x = torch.nn.functional.normalize(x, dim=1)
-        return x
+# class SiameseModel(nn.Module):
+#     def __init__(self):
+#         super().__init__()
+#         self.backbone = InceptionResnetV1(pretrained='vggface2')
+#     def forward(self, x):
+#         x = self.backbone(x)
+#         x = torch.nn.functional.normalize(x, dim=1)
+#         return x
     
-model = SiameseModel()
-model.load_state_dict(torch.load('model.pt', map_location=torch.device('cpu')))
-model.eval()
+# model = SiameseModel()
+# model.load_state_dict(torch.load('model.pt', map_location=torch.device('cpu')))
+# model.eval()
 
 
 # Make FAISS index