Update app.py

app.py

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