Update app.py

app.py

@@ -5,6 +5,9 @@ import spaces
 import torch
 from diffusers import DiffusionPipeline
 
+# for initial data
+pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=dtype).to("cpu")
+
 dtype = torch.bfloat16
 #device = "cuda" if torch.cuda.is_available() else "cpu"
 device = "cuda"
@@ -46,7 +49,7 @@ css="""
 with gr.Blocks(css=css) as demo:
  
  with gr.Column(elem_id="col-container"):
- gr.Markdown(f"""# FLUX.1 [schnell]
+ gr.Markdown(f"""# FLUX.1 [schnell] 120 sec version
 12B param rectified flow transformer distilled from [FLUX.1 [pro]](https://blackforestlabs.ai/) for 4 step generation
 [[blog](https://blackforestlabs.ai/announcing-black-forest-labs/)] [[model](https://huggingface.co/black-forest-labs/FLUX.1-schnell)]
  """)