diff --git a/run.py b/run.py
index 1a2c5c9..e3cbdef 100644
--- a/run.py
+++ b/run.py
@@ -1,11 +1,11 @@
 import os
-os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
-os.environ["NO_ALBUMENTATIONS_UPDATE"] = "1"
 import sys
 from typing import Union, OrderedDict
 from dotenv import load_dotenv
 # Load the .env file if it exists
 load_dotenv()
+os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = os.getenv("HF_HUB_ENABLE_HF_TRANSFER", "1")
+os.environ["NO_ALBUMENTATIONS_UPDATE"] = "1"
 
 sys.path.insert(0, os.getcwd())
 # must come before ANY torch or fastai imports