Update app.py

app.py

@@ -1,6 +1,6 @@
 import torch
 import gradio as gr
-from diffusers import StableDiffusionXLPipeline
+from diffusers import DiffusionPipeline
 from PIL import Image
 from io import BytesIO
 import os
@@ -8,13 +8,9 @@ import requests
 import time
 from tqdm import tqdm
 
-# Load local Stable Diffusion XL model
-model_path = "networks/TShirtDesignRedmondV2-Tshirtdesign-TshirtDesignAF.safetensors"
-pipe = StableDiffusionXLPipeline.from_single_file(
-    model_path,
-    torch_dtype=torch.float16, 
-    use_safetensors=True,
-)
+# Load Stable Diffusion XL base model with LoRA weights
+pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
+pipe.load_lora_weights("networks/TShirtDesignRedmondV2-Tshirtdesign-TshirtDesignAF.safetensors")
 pipe = pipe.to("cuda")
 
 def infer(color_prompt, dress_type_prompt, design_prompt):