Update doors_fasterrcnn.py

doors_fasterrcnn.py

@@ -407,7 +407,7 @@ def main_run(img_pillow,pdf_fullpath, weights_path, weights_path2, pdf_name,pdfp
     # Load the model with the specified number of classes
     model2 = get_model(num_classes2)
     # 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')
+    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)
     except RuntimeError as e:
@@ -416,11 +416,11 @@ def main_run(img_pillow,pdf_fullpath, weights_path, weights_path2, pdf_name,pdfp
     # Set the model to evaluation mode
     model2.eval()
     # Move the model to the appropriate device
-    model2.to(device)
+    model2.to(device2)
     
     # START INFERENCE
     doors_info = ev_model(img_pillow, model, device, 0.9)
-    doors_info_double = ev_model(img_pillow, model2, device, 0.8)
+    doors_info_double = ev_model(img_pillow, model2, device2, 0.8)
 
     
     width_pixels, lines, sanda, line_midpoint, single_count, door_type = get_door_info(doors_info)