Update app.py

app.py

@@ -2,16 +2,17 @@ import gradio as gr
 import torch
 from diffusers import KolorsPipeline
 
-pipeline= KolorsPipeline.from_pretrained(
-    "Kwai-Kolors/Kolors-diffusers", 
-    torch_dtype=torch.float16, 
-    variant="fp16"
-).to("cuda" if torch.cuda.is_available() else "cpu")
+from diffusers import FluxPipeline
+
+pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
+pipe.load_lora_weights("Shakker-Labs/FLUX.1-dev-LoRA-Logo-Design", weight_name="FLUX-dev-lora-Logo-Design.safetensors")
+pipe.fuse_lora(lora_scale=0.8)
+pipe.to("cuda" if torch.cuda.is_available() else "cpu")
 
 # Function to generate image from prompt
 def generate_image(prompt):
     # Use the pipeline to generate an image from the text prompt
-    image = pipeline(prompt).images[0]
+    image = pipe(prompt).images[0]
     return image
 
 # Create Gradio interface