Update app.py

app.py

@@ -11,9 +11,11 @@ import os
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
+# Set your Hugging Face API token
 huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
 
-pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=dtype, token = huggingface_token).to(device)
+# Load the diffusion pipeline with the Hugging Face API token
+pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=dtype, token=huggingface_token).to(device)
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
@@ -97,7 +99,6 @@ body {
 
 with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="blue", secondary_hue="gray")) as demo:
     
-    
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""# FLUX.1 [dev]
 12B param rectified flow transformer guidance-distilled from FLUX.1 [pro]
@@ -110,7 +111,6 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="blue", secondary_hue="
 </a>
         """)
 
-            
         with gr.Row():
             prompt = gr.Text(
                 label="Prompt",
@@ -189,7 +189,6 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="blue", secondary_hue="
             cache_examples="lazy"
         )
 
-
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=infer,
@@ -203,5 +202,4 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="blue", secondary_hue="
         outputs=None
     )
 
-
 demo.launch()