Update demo_utils.py

demo_utils.py

@@ -61,7 +61,11 @@ def run_demo(dataset_name, prompt_type, sample_idx, prompt_idx, voxel_size, thet
        if not args.ret_prompt: return new_color, PROMPT
        else:   return PROMPT
    except Exception as e:
-       pass
+       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 args.dataset == 'S3DIS':
        info = configs.S3DIS_samples[args.sample_idx]