multimodalart HF staff commited on
Commit
349bdb0
1 Parent(s): a08da6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -13,6 +13,11 @@ import copy
13
  import random
14
  import time
15
  import requests
 
 
 
 
 
16
 
17
  # Load LoRAs from JSON file
18
  with open('loras.json', 'r') as f:
@@ -191,7 +196,8 @@ def randomize_loras(selected_indices):
191
  lora_scale_2 = 0.95
192
  lora_image_1 = lora1['image']
193
  lora_image_2 = lora2['image']
194
- return selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2
 
195
 
196
  @spaces.GPU(duration=70)
197
  def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress):
@@ -570,7 +576,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, delete_cache=(60, 3600)) as app:
570
  randomize_button.click(
571
  randomize_loras,
572
  inputs=[selected_indices],
573
- outputs=[selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2]
574
  )
575
  add_custom_lora_button.click(
576
  add_custom_lora,
 
13
  import random
14
  import time
15
  import requests
16
+ import pandas as pd
17
+
18
+ #Load prompts for randomization
19
+ df = pd.read_csv('prompts.csv', header=None)
20
+ prompt_values = df.values.flatten()
21
 
22
  # Load LoRAs from JSON file
23
  with open('loras.json', 'r') as f:
 
196
  lora_scale_2 = 0.95
197
  lora_image_1 = lora1['image']
198
  lora_image_2 = lora2['image']
199
+ random_prompt = random.choice(prompt_values)
200
+ return selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2, random_prompt
201
 
202
  @spaces.GPU(duration=70)
203
  def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress):
 
576
  randomize_button.click(
577
  randomize_loras,
578
  inputs=[selected_indices],
579
+ outputs=[selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2, prompt]
580
  )
581
  add_custom_lora_button.click(
582
  add_custom_lora,