Update doors_fasterrcnn.py

doors_fasterrcnn.py

@@ -409,7 +409,7 @@ def main_run(img_pillow,pdf_fullpath, weights_path, weights_path2, pdf_name,pdfp
     # Load the saved model's state dictionary with map_location to handle CPU
     device2 = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
     try:
-        model.load_state_dict(torch.load(weights_path2, map_location=device), strict=False)
+        model2.load_state_dict(torch.load(weights_path2, map_location=device2), strict=False)
     except RuntimeError as e:
         print(f"Error loading model state_dict: {e}")
         return