Update app.py

app.py

@@ -69,6 +69,9 @@ class ImageClassifierCataract:
         object_box_color = (255, 0, 0)
         cv2.rectangle(output_image, (box_x, box_y), (box_x + box_size, box_y + box_size), object_box_color, 2)
 
+        # Crop the output image to keep only the region of interest
+        output_image = output_image[box_y:box_y+box_size, box_x:box_x+box_size]
+
         return output_image
 
     def run_interface(self):
@@ -83,4 +86,4 @@ class ImageClassifierCataract:
 if __name__ == "__main__":
     model_path = 'modelo_treinado.h5'  # Replace with the path to your trained model
     app = ImageClassifierCataract(model_path)
-    app.run_interface()
+    app.run_interface()