Update app_docker.py

app_docker.py

@@ -11,10 +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)
+# 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