Update demo_utils.py

demo_utils.py

@@ -62,8 +62,8 @@ def run_demo(dataset_name, prompt_type, sample_idx, prompt_idx, voxel_size, thet
        if os.path.exists("./cache_results/" + name + '.npy') and os.path.exists("./cache_prompt/" + name + '.npy'):
            new_color = np.load("./cache_results/" + name + '.npy')
            PROMPT = np.load("./cache_prompt/" + name + '.npy')
-       if not args.ret_prompt: return new_color, PROMPT
-       else:   return PROMPT
+           if not args.ret_prompt: return new_color, PROMPT
+           else:   return PROMPT
    #########
    if args.dataset == 'S3DIS':
        info = configs.S3DIS_samples[args.sample_idx]