Update doors_fasterrcnn.py

doors_fasterrcnn.py

@@ -176,7 +176,7 @@ def main_run(pdf_fullpath, weights_path, pdf_name):
   # Load the saved model's state dictionary with map_location to handle CPU
   device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
   # model.load_state_dict(torch.load(weights_path, map_location=device))
-  model.load_state_dict(torch.load(weights_path, map_location=torch.device('cpu')))
+  model.load_state_dict(torch.load(weights_path, map_location=device), strict=False)
   # Set the model to evaluation mode
   model.eval()
   # Move the model to the appropriate device