Update app.py

app.py

@@ -4,9 +4,10 @@ from diffusers import FluxPipeline
 from PIL import Image
 
 # Load the diffusion model
-pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
+pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16)
 pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
 
+
 # Set the model to the appropriate device
 #device = "cuda" if torch.cuda.is_available() else "cpu"
 #pipeline.to(device)