Update app.py

app.py

@@ -5,6 +5,7 @@ import spaces
 import torch
 from diffusers import StableDiffusionXLPipeline, EulerDiscreteScheduler
 from compel import Compel, ReturnedEmbeddingsType
+from huggingface_hub import hf_hub_download
 
 import re
 
@@ -224,9 +225,15 @@ if not torch.cuda.is_available():
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 4096
 
+local_ckpt_path = hf_hub_download(
+    repo_id="Menyu/ModelFile", 
+    filename="noobaiXLNAIXL_vPred10Version.safetensors", 
+    revision="main"
+)
+
 if torch.cuda.is_available():
     pipe = StableDiffusionXLPipeline.from_pretrained(
-        "Menyu/noobaiXLNAIXL_vPred10Version",
+        local_ckpt_path,
         torch_dtype=torch.float16
     )
     scheduler_args = {"prediction_type": "v_prediction", "rescale_betas_zero_snr": True}