Update app.py

app.py

@@ -2,16 +2,18 @@ import torch
 import gradio as gr
 from diffusers import DiffusionPipeline
 from PIL import Image
-from io import BytesIO
-import os
-import requests
-import time
-from tqdm import tqdm
 
-# Load Stable Diffusion XL base model with LoRA weights
+# Load Stable Diffusion XL base model with LoRA weights (requires PEFT backend)
 pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
-pipe.load_lora_weights("networks/TShirtDesignRedmondV2-Tshirtdesign-TshirtDesignAF.safetensors")
-pipe = pipe.to("cuda")
+pipe.to("cuda")
+
+# Ensure PEFT is installed and compatible
+try:
+    from peft import LoraModel
+    pipe.load_lora_weights("networks/TShirtDesignRedmondV2-Tshirtdesign-TshirtDesignAF.safetensors")
+except ImportError:
+    raise ImportError("PEFT is required for loading LoRA weights. Install it using `pip install peft`." )
+
 
 def infer(color_prompt, dress_type_prompt, design_prompt):
     prompt = (