Update app.py

app.py

@@ -11,6 +11,9 @@ import spaces
 import torch
 from diffusers import StableDiffusionPipeline, StableDiffusionInpaintPipeline, StableDiffusionXLPipeline, StableDiffusionXLInpaintPipeline, EulerAncestralDiscreteScheduler, DPMSolverSinglestepScheduler
 from diffusers.utils import load_image
+from huggingface_hub import hf_hub_download
+
+HF_TOKEN = os.getenv("HF_TOKEN")
 
 DESCRIPTION = """
 # [Fluently Playground](https://huggingface.co/fluently)
@@ -29,8 +32,8 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 
 if torch.cuda.is_available():
-    pipe_xl_final = StableDiffusionXLPipeline.from_pretrained(
-        "fluently/Fluently-XL-Final",
+    pipe_xl_final = StableDiffusionXLPipeline.from_single_file(
+        hf_hub_download(repo_id="fluently/Fluently-XL-Final", filename="FluentlyXL-Final.safetensors", token=HF_TOKEN),
         torch_dtype=torch.float16,
         use_safetensors=True,
     )