Update app.py

app.py

@@ -10,7 +10,14 @@ schnell_model = "black-forest-labs/FLUX.1-schnell"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 pipe_dev = DiffusionPipeline.from_pretrained(dev_model, torch_dtype=torch.bfloat16).to(device)
-pipe_schnell = DiffusionPipeline.from_pretrained(schnell_model, torch_dtype=torch.bfloat16).to(device)
+pipe_schnell = DiffusionPipeline.from_pretrained(
+    schnell_model,
+    text_encoder=pipe_dev.text_encoder,
+    text_encoder_2=pipe_dev.text_encoder_2,
+    tokenizer=pipe_dev.tokenizer,
+    tokenizer_2=pipe_dev.tokenizer_2,
+    torch_dtype=torch.bfloat16
+).to(device)
 
 @spaces.GPU
 def run_dev_hyper(prompt):