add token

app.py

@@ -23,9 +23,16 @@ def init_pipeline():
         use_auth_token=os.getenv("HF_TOKEN"),
     )
     pipe = FluxConditionalPipeline.from_pretrained(
-        "black-forest-labs/FLUX.1-dev", transformer=transformer, torch_dtype=torch.bfloat16, use_auth_token=os.getenv("HF_TOKEN")
+        "black-forest-labs/FLUX.1-dev",
+        transformer=transformer,
+        torch_dtype=torch.bfloat16,
+        use_auth_token=os.getenv("HF_TOKEN"),
+    )
+    pipe.load_lora_weights(
+        pretrained_model_name_or_path="primecai/dsd_model",
+        weight_name="pytorch_lora_weights.safetensors",
+        use_auth_token=os.getenv("HF_TOKEN"),
     )
-    pipe.load_lora_weights(os.path.join(CHECKPOINT, "pytorch_lora_weights.safetensors"), use_auth_token=os.getenv("HF_TOKEN"))
     pipe.to("cuda")