Update mast3r.py

mast3r.py

@@ -8,7 +8,7 @@ from mini_dust3r.utils.misc import (
 from mini_dust3r.model import load_model
 from pathlib import Path
 import uuid
-from utils import create_image_grid  # Corrected import
+from utils import create_image_grid
 import rerun as rr
 import rerun.blueprint as rrb
 import os
@@ -40,7 +40,7 @@ class AsymmetricMASt3R(AsymmetricCroCo3DStereo):
         # allocate heads
         from mini_dust3r.heads.linear_head import LinearPts3d
         from mini_dust3r.heads.dpt_head import create_dpt_head
-        from catmlp_dpt_head import Cat_MLP_LocalFeatures_DPT_Pts3d, postprocess
+        from catmlp_dpt_head import Cat_MLP_LocalFeatures_DPT_Pts3d, postprocess # Corrected import
 
         def head_factory(head_type, output_mode, net, has_conf=False):
             """" build a prediction head for the decoder
@@ -113,6 +113,7 @@ class MASt3R:
             )
         return blueprint
 
+
     def generate_point_cloud(self, image_name_list):
         if not isinstance(image_name_list, list) and not isinstance(image_name_list, str):
             raise Exception(