Spaces:

malmukhtar
/

Accessibility-labeler

Sleeping

malmukhtar commited on Aug 2, 2023

Commit

4ca28a0

1 Parent(s): 2283d5f

Update accessibility_classifier/classifier/model.py

Files changed (1) hide show

accessibility_classifier/classifier/model.py CHANGED Viewed

@@ -68,7 +68,7 @@ class Model:
         )
         classifier.load_state_dict(
-            torch.load("assets/pytorch_model.bin", map_location=self.device))
         classifier = classifier.eval()
         self.classifier = classifier.to(self.device)

         )
         classifier.load_state_dict(
+            torch.load("assets/pytorch_model.bin", map_location=self.device), strict=False)
         classifier = classifier.eval()
         self.classifier = classifier.to(self.device)