Update main.py

main.py

@@ -3,6 +3,11 @@ import os
 os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/huggingface_cache"
 os.makedirs("/tmp/huggingface_cache", exist_ok=True)
 
+os.environ["NUMBA_CACHE_DIR"] = "/tmp/numba_cache"
+
+# Create the directory
+os.makedirs("/tmp/numba_cache", exist_ok=True)
+
 import torch
 import numpy as np
 from PIL import Image