Update app.py

app.py

@@ -2,14 +2,17 @@ import gradio as gr
 import torch
 import spaces
 from diffusers import FluxPipeline
+import os
 
 # 1. Initialize the Pipeline
+hf_token = os.getenv("HF_TOKEN")
 model_id = "black-forest-labs/FLUX.1-schnell"
 
 # Load the model with bfloat16 to save memory and improve speed
 pipe = FluxPipeline.from_pretrained(
     model_id, 
     torch_dtype=torch.bfloat16
+    hf_token = os.getenv("FLUX_API_TOKEN")
 )
 
 # 2. Define the Generation Function with ZeroGPU decorator