add app

app.py

@@ -9,7 +9,7 @@ import spaces #[uncomment to use ZeroGPU]
 import torch
 from torchvision.transforms import ToTensor, ToPILImage
 import logging
-logging.getLogger("huggingface_hub").setLevel(logging.CRITICAL)
+# logging.getLogger("huggingface_hub").setLevel(logging.CRITICAL)
 from huggingface_hub import hf_hub_download, snapshot_download
 
 model_name = "iimmortall/UltraFusion"
@@ -17,11 +17,10 @@ auth_token = os.getenv("HF_AUTH_TOKEN")
 # print(auth_token)
 # greet_file = hf_hub_download(repo_id=model_name, filename="main.py", use_auth_token=auth_token) 
 # sys.path.append(os.path.split(greet_file)[0])  
-model_folder = snapshot_download(repo_id=model_name, token=auth_token)
+model_folder = snapshot_download(repo_id=model_name, token=auth_token, local_dir="/home/user/app")
 # sys.path.append(model_folder)
-sys.path.insert(0, model_folder)
-print(sys.path)
-# exit()
+# sys.path.insert(0, model_folder)
+# print(sys.path)
 
 from ultrafusion_utils import load_model, run_ultrafusion