Update main.py

main.py

@@ -16,7 +16,7 @@ MAX_SEED = np.iinfo(np.int32).max
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 # Load pipelines
-HF_TOKEN = "your_huggingface_token"  # Replace with your actual token
+HF_TOKEN = os.getenv("HF_TOKEN") # Replace with your actual token
 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,