Update app.py

app.py

@@ -1,22 +1,10 @@
 import gradio as gr
 import spaces
-from diffusers import FluxPipeline
+from diffusers import DiffusionPipeline
 import torch
 
-# Replace with your actual model and LoRA paths
-BASE_MODEL = "black-forest-labs/FLUX-1-dev"
-LORA_MODEL = "MegaTronX/MetartLoRA" # Replace with your repo
-
-# Load base model pipeline
-pipe = FluxPipeline.from_pretrained(BASE_MODEL, torch_dtype=torch.float16)
-
-#pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev")
-#pipe.load_lora_weights("MegaTronX/MetartLoRA")
-
-
-# Load LoRA weights
-pipe.unet.load_attn_procs(LORA_MODEL)
-pipe = pipe.to("cuda")
+pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev")
+pipe.load_lora_weights("MegaTronX/MetartLoRA")
 
 @spaces.GPU
 def generate_image(prompt, num_inference_steps=25, guidance_scale=7.5, seed=None):