diff --git a/generate_equirect.py b/generate_equirect.py
index 23920fd..33d22ab 100644
--- a/generate_equirect.py
+++ b/generate_equirect.py
@@ -12,6 +12,8 @@ import json
 import math
 import os
 import re
+import shutil
+import sys
 import tempfile
 import torch
 from PIL import Image, ImageDraw