Create app.py

app.py

@@ -0,0 +1,32 @@
+import torch
+import gradio as gr
+from diffusers import StableDiffusionXLPipeline, DPMSolverMultistepScheduler
+from peft import PeftModel, PeftConfig
+
+# Load SDXL base pipeline
+base_model_id = "stabilityai/stable-diffusion-xl-base-1.0"
+pipe = StableDiffusionXLPipeline.from_pretrained(
+    base_model_id,
+    torch_dtype=torch.float16,
+    variant="fp16",
+    use_safetensors=True
+).to("cuda")
+
+# Load LoRA weights
+lora_path = "./DreamCartoonLora.safetensors"
+pipe.load_lora_weights(lora_path)
+
+# Optional: set scheduler for faster/cleaner results
+pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
+
+def generate(prompt):
+    image = pipe(prompt=prompt).images[0]
+    return image
+
+gr.Interface(
+    fn=generate,
+    inputs=gr.Textbox(label="Enter your prompt"),
+    outputs="image",
+    title="Dream Cartoon LoRA - SDXL 1.0",
+    description="Generate images using the DreamCartoonLora fine-tuned on SDXL 1.0"
+).launch()