Update app.py

app.py

@@ -4,15 +4,18 @@ import random
 from PIL import Image
 import spaces
 import torch
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download, HfApi
 from diffusers import FluxPriorReduxPipeline, FluxPipeline
 from diffusers.utils import load_image
-
+import os
+api = HfApi(
+    token=os.getenv('HF_TOKEN'), # Token is not persisted on the machine.
+)
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 
 pipe = FluxPipeline.from_pretrained(
-    "black-forest-labs/FLUX.1-dev" , 
+    "LPX55/FLUX.1-merged_uncensored", 
     torch_dtype=torch.bfloat16
 ).to("cuda")
 pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"), lora_scale=0.125)