Update app.py

app.py

@@ -8,8 +8,14 @@ def load_pipeline():
     lora_repo = "strangerzonehf/Flux-Super-Realism-LoRA"
     trigger_word = "Super Realism"  # Recommended trigger word
 
-    # Load the base model
-    pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
+# Retrieve your Hugging Face token from an environment variable
+hf_token = os.environ.get("HF_TOKEN")
+
+pipe = DiffusionPipeline.from_pretrained(
+    "black-forest-labs/FLUX.1-dev",
+    torch_dtype=torch.bfloat16,
+    use_auth_token=hf_token  # Use the token stored in your environment variable
+)
 
     # Load the LoRA weights into the pipeline
     pipe.load_lora_weights(lora_repo)