Update inference.py

inference.py

@@ -34,17 +34,22 @@ class Inference():
                 raise ValueError(f"Error processing image {idx}: {str(e)}")
 
         return prepared_images
-
-    def classify_building(self, images: List[Image.Image]) -> dict:
+        
+    
+    def classify_building(self, images: List[Image.Image], saved_image_paths: List[str] = None) -> dict:
         """
         Classify building type from a list of PIL Image objects
         Args:
             images: List of PIL Image objects
+            saved_image_paths: List of paths where images were saved to disk (optional)
         Returns:
             Classification response
         """
         logger.info(f"Preparing {len(images)} images for classification")
+        if saved_image_paths:
+            logger.info(f"Images saved to disk at: {saved_image_paths}")
+        
         prepared_images = self._prepare_images(images)
         logger.info(f"Image preparation successful")
-        response = self.classifier.get_response(prepared_images)
+        response = self.classifier.get_response(prepared_images, saved_image_paths)
         return response