Update app.py

app.py

@@ -86,9 +86,9 @@ div#share-btn-container > div {flex-direction: row;background: black;align-items
 #thumbs_down_clicked{background:red}
 .title_lora a{color: var(--body-text-color) !important; opacity:0.6}
 #prompt_area .form{border:0}
+#reroll_button{position: absolute;right: 0;flex-grow: 1;min-width: 75px;padding: .1em}
 '''
 
-
 original_pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
 
 @spaces.GPU
@@ -104,7 +104,6 @@ def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lor
   state_dict_2 = {k: v.to(device="cuda", dtype=torch.float16) for k,v in state_dict_2.items() if torch.is_tensor(v)}
   state_dict_time = time() - start_time
   print(f"State Dict time: {state_dict_time}")  
-  #pipe = copy.deepcopy(original_pipe)
   start_time = time()
   unet = copy.deepcopy(original_pipe.unet)
   text_encoder=copy.deepcopy(original_pipe.text_encoder)
@@ -134,15 +133,12 @@ def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lor
       seed = random.randint(0, 2147483647)
   generator = torch.Generator(device="cuda").manual_seed(seed)
   image = pipe(prompt=prompt, negative_prompt=negative_prompt, num_inference_steps=20, width=768, height=768, generator=generator).images[0]
-  del pipe
-  gc.collect()
   return image, gr.update(visible=True), seed, gr.update(visible=True, interactive=True), gr.update(visible=False), gr.update(visible=True, interactive=True), gr.update(visible=False)
 
 def get_description(item):
       trigger_word = item["trigger_word"]
       return f"Trigger: `{trigger_word}`" if trigger_word else "No trigger, applied automatically", trigger_word
 
-
 def truncate_string(s, max_length=29):
     return s[:max_length - 3] + "..." if len(s) > max_length else s
 
@@ -179,7 +175,7 @@ def save_preferences(lora_1_id, lora_1_scale, lora_2_id, lora_2_scale, prompt, g
 with gr.Blocks(css=css) as demo:
   shuffled_items = gr.State()
   title = gr.HTML(
-        '''<h1>LoRA Roulette 🎲</h1>
+        '''<h1>LoRA Roulette </h1>
         <p>This random LoRAs are loaded into SDXL, can you find a fun way to combine them? 🎨</p>
         ''',
         elem_id="title"
@@ -201,6 +197,7 @@ with gr.Blocks(css=css) as demo:
               lora_2_prompt = gr.Markdown(visible=False)
             with gr.Column(min_width=10, scale=2, elem_classes="plus_column"):
                equal = gr.HTML("=", elem_classes="plus_button")
+            shuffle_button = gr.Button("🎲  reroll", elem_id="reroll_button")
     with gr.Column(min_width=10, scale=14):
         with gr.Box(elem_id="generate_area"):
             with gr.Row(elem_id="prompt_area"):
@@ -226,7 +223,6 @@ with gr.Blocks(css=css) as demo:
     with gr.Row():
       lora_1_scale = gr.Slider(label="LoRA 1 Scale", minimum=0, maximum=1, step=0.1, value=0.7)
       lora_2_scale = gr.Slider(label="LoRa 2 Scale", minimum=0, maximum=1, step=0.1, value=0.7)
-  shuffle_button = gr.Button("Reshuffle!")
   gr.Markdown("For generating with intent visit the [LoRA the Explorer Space](https://huggingface.co/spaces/multimodalart/LoraTheExplorer), but don't forget that sometimes restirctions flourish creativity 🌸")
     
   demo.load(shuffle_images, inputs=[], outputs=[lora_1_link, lora_1, lora_1_prompt, lora_1_id, lora_2_link, lora_2, lora_2_prompt, lora_2_id, prompt, shuffled_items, lora_1_scale, lora_2_scale], queue=False, show_progress="hidden")