Update app.py

app.py

@@ -11,8 +11,10 @@ summarizer = pipeline("summarization", model="Falconsai/text_summarization")
 # image_pipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16).to(device)
 # image_pipe = fluxPipeline("image-generation", model=DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16).to(device))
 
-image_pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16).to(device)
-image_pipe.enable_model_cpu_offload()
+# image_pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16).to(device)
+# image_pipe.enable_model_cpu_offload()
+image_pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev")
+image_pipe.to(device)
 
 
 # Functions for each task