Update doors_fasterrcnn.py

doors_fasterrcnn.py

@@ -175,8 +175,8 @@ def main_run(pdf_fullpath, weights_path, pdf_name):
   model = get_model(num_classes)
   # 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=device), strict=False)
+  model.load_state_dict(torch.load(weights_path, map_location=device))
+  # 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