update model download method

app.py

@@ -1,5 +1,4 @@
 import os
-os.system("pip install git+https://${GIT_USER}:${GIT_TOKEN}@github.com/KohakuBlueleaf/XUT")
 import random
 import json
 from pathlib import Path
@@ -9,6 +8,7 @@ import httpx
 if os.environ.get("IN_SPACES", None) is not None:
     in_spaces = True
     import spaces
+    os.system("pip install git+https://${GIT_USER}:${GIT_TOKEN}@github.com/KohakuBlueleaf/XUT")
 else:
     in_spaces = False
 import numpy as np
@@ -68,7 +68,7 @@ def download_model(url: str, filepath: str):
         print(f"Model already exists at {filepath}")
         return
     
-    print(f"Downloading model from {url}...")
+    print(f"Downloading model...")
     Path(filepath).parent.mkdir(parents=True, exist_ok=True)
     
     with httpx.stream("GET", url, follow_redirects=True) as response:
@@ -168,8 +168,8 @@ vae_std = torch.tensor(vae.config.latents_std).view(1, -1, 1, 1).to(device)
 
 
 if not os.path.exists("./model/model.safetensors"):
-    model_url = os.environ.get("MODEL_URL")
-    download_model(model_url, "./model/model.safetensors")
+    model_file = os.environ.get("MODEL_FILE")
+    os.system(f"hfutils download -t model -r KBlueLeaf/XUT-demo -f {model_file} -o model/model.safetensors")
 
 state_dict = load_file("./model/model.safetensors")
 model_sd = {k.replace("unet.", ""): v for k, v in state_dict.items() if k.startswith("unet.")}

requirements.txt

@@ -9,4 +9,5 @@ spaces
 gradio
 httpx
 einops
+hfutils[transfer]
 https://github.com/abetlen/llama-cpp-python/releases/download/v0.3.13-cu124/llama_cpp_python-0.3.13-cp310-cp310-linux_x86_64.whl