Update app_docker.py

app_docker.py

@@ -11,13 +11,13 @@ from torchvision.transforms import ToTensor, ToPILImage
 
 
 # -------------------------- HuggingFace -------------------------------
-# from huggingface_hub import hf_hub_download, snapshot_download
-# model_name = "iimmortall/UltraFusion"
-# auth_token = os.getenv("HF_AUTH_TOKEN")
-# model_folder = snapshot_download(repo_id=model_name, token=auth_token, local_dir="/home/user/app", force_download=True)
-import sys
-sys.path.append("/data")
-sys.path.append("/data/")
+from huggingface_hub import hf_hub_download, snapshot_download
+model_name = "iimmortall/UltraFusion"
+auth_token = os.getenv("HF_AUTH_TOKEN")
+model_folder = snapshot_download(repo_id=model_name, token=auth_token, local_dir="/home/user/app", force_download=True)
+# import sys
+# sys.path.append("/data")
+# sys.path.append("/data/")
 
 from ultrafusion_utils import load_model, run_ultrafusion, check_input
 PYCUDA_FLAG = True