Update handler.py

handler.py

@@ -72,8 +72,8 @@ class EndpointHandler():
         if use_image:
             sampler = PointCloudSampler(
                 device=device,
-                models=[base_model, upsampler_model],
-                diffusions=[base_diffusion, upsampler_diffusion],
+                models=[self.base_image_model, self.upsampler_model],
+                diffusions=[self.base_diffusion, self.upsampler_diffusion],
                 num_points=[1024, 4096 - 1024],
                 aux_channels=['R', 'G', 'B'],
                 guidance_scale=[3.0, 3.0],