Update app.py (#1)

- Update app.py (a0e676cdd023bbb9fbeba5c2120a2fa264992186)


Co-authored-by: Ben Greer <[email protected]>

app.py

@@ -13,8 +13,6 @@ from diffusers import StableDiffusionPipeline, StableDiffusionInpaintPipeline, S
 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)
 
@@ -33,7 +31,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 if torch.cuda.is_available():
     pipe_xl_final = StableDiffusionXLPipeline.from_single_file(
-        hf_hub_download(repo_id="fluently/Fluently-XL-Final", filename="FluentlyXL-Final.safetensors", token=HF_TOKEN),
+        hf_hub_download(repo_id="fluently/Fluently-XL-Final", filename="FluentlyXL-Final.safetensors"),
         torch_dtype=torch.float16,
         use_safetensors=True,
     )
@@ -56,11 +54,16 @@ if torch.cuda.is_available():
     pipe_epic.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe_epic.scheduler.config)
     pipe_epic.to(device)
 
-    pipe_xl_inpaint = StableDiffusionXLInpaintPipeline.from_single_file(
-        "https://huggingface.co/fluently/Fluently-XL-v3-inpainting/blob/main/FluentlyXL-v3-inpainting.safetensors",
+    pipe_xl_inpaint = StableDiffusionXLInpaintPipeline.from_pretrained(
+        "fluently/Fluently-XL-v3-inpainting",
         torch_dtype=torch.float16,
-        use_safetensors=True,
-    )
+        use_safetensors=True,)
+
+    # pipe_xl_inpaint = StableDiffusionXLInpaintPipeline.from_single_file(
+    #     "https://huggingface.co/fluently/Fluently-XL-v3-inpainting/blob/main/FluentlyXL-v3-inpainting.safetensors",
+    #     torch_dtype=torch.float16,
+    #     use_safetensors=True,
+    # )
     pipe_xl_inpaint.to(device)
 
     pipe_inpaint = StableDiffusionInpaintPipeline.from_pretrained(