Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from time import sleep, time
-from diffusers import DiffusionPipeline
+from diffusers import DiffusionPipeline, StableDiffusionXLPipeline
 from huggingface_hub import hf_hub_download, CommitScheduler
 from safetensors.torch import load_file
 from share_btn import community_icon_html, loading_icon_html, share_js
@@ -87,6 +87,7 @@ div#share-btn-container > div {flex-direction: row;background: black;align-items
 .title_lora a{color: var(--body-text-color) !important}
 '''
 
+
 original_pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
 
 @spaces.GPU
@@ -102,8 +103,8 @@ def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lor
   print(f"State Dict time: {state_dict_time}")  
   #pipe = copy.deepcopy(original_pipe)
   start_time = time()
-  pipe = original_pipe
-  pipe.unet = copy.deepcopy(original_pipe.unet)
+  unet = copy.deepcopy(original_pipe.unet)
+  pipe = StableDiffusionXLPipeline(vae=original_pipe.vae, text_encoder=original_pipe.text_encoder, text_encoder_2=original_pipe.text_encoder_2, scheduler=original_pipe.scheduler, unet=unet)
   pickle_time = time() - start_time
   print(f"copy time: {pickle_time}")  
   pipe.to("cuda")