Enhance image loading function with additional debug prints and improved error logging

models.py

@@ -102,13 +102,15 @@ def load_image(image_file, input_size=448, max_num=12):
         image = correct_image_orientation(image_file).convert('RGB')
         print("Image size:", image.size)
         transform = build_transform(input_size=input_size)
+        print("Transform built.")
         images = dynamic_preprocess(image, image_size=input_size, use_thumbnail=True, max_num=max_num)
         print("Number of images:", len(images))
         pixel_values = [transform(image) for image in images]
+        print("Images transformed.")
         pixel_values = torch.stack(pixel_values)
         print("Image loaded successfully.")
     except Exception as e:
         print("Error loading image:", e)
-        traceback.print_exc()
+        print(traceback.format_exc())
         pixel_values = None
     return pixel_values