updated

app.py

@@ -5,7 +5,7 @@ import spaces
 from huggingface_hub import hf_hub_download
 from utils import utils, tools, preprocess
 
-# BASE_MODEL_PATH = "stablediffusionapi/neta-art-xl-v2"
+BASE_MODEL_PATH = "stablediffusionapi/neta-art-xl-v2"
 VAE_PATH = "madebyollin/sdxl-vae-fp16-fix"
 REPO_ID = "Pbihao/ControlNeXt"
 UNET_FILENAME = "ControlAny-SDXL/anime_canny/unet.safetensors"
@@ -18,16 +18,6 @@ DEFAULT_NEGATIVE_PROMPT = "worst quality, abstract, clumsy pose, deformed hand,
 
 def ui():
     device = "cuda" if torch.cuda.is_available() else "cpu"
-    # model_file = hf_hub_download(
-    #     repo_id='Lykon/AAM_XL_AnimeMix',
-    #     filename='AAM_XL_Anime_Mix.safetensors',
-    #     cache_dir=CACHE_DIR,
-    # )
-    model_file = hf_hub_download(
-        repo_id='Eugeoter/artiwaifu-diffusion-1.0',
-        filename='artiwaifu-diffusion-v1.safetensors',
-        cache_dir=CACHE_DIR,
-    )
     unet_file = hf_hub_download(
         repo_id=REPO_ID,
         filename=UNET_FILENAME,
@@ -40,7 +30,7 @@ def ui():
     )
 
     pipeline = tools.get_pipeline(
-        pretrained_model_name_or_path=model_file,
+        pretrained_model_name_or_path=BASE_MODEL_PATH,
         unet_model_name_or_path=unet_file,
         controlnet_model_name_or_path=controlnet_file,
         vae_model_name_or_path=VAE_PATH,