Update app.py

app.py

@@ -1,8 +1,9 @@
 import gradio as gr
 from transformers import pipeline
+from diffusers import FluxPipeline
 
-pipe = pipeline("image-to-text",
-                model="Salesforce/blip-image-captioning-base")
+FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
+pipe.enable_model_cpu_offload()
 
 
 def launch(input):