svjack commited on
Commit
090dbc6
1 Parent(s): f34cbc2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -15,7 +15,8 @@ prompt_df = dataset["train"].to_pandas()
15
  prompt_df = pd.read_csv("Stable-Diffusion-Prompts.csv")
16
 
17
  #DEFAULT_MODEL = "stabilityai/stable-diffusion-2-1"
18
- DEFAULT_PROMPT = "1girl, aqua eyes, baseball cap, blonde hair, closed mouth, earrings, green background, hat, hoop earrings, jewelry, looking at viewer, shirt, short hair, simple background, solo, upper body, yellow shirt"
 
19
 
20
  def get_samples():
21
  prompt_list = prompt_df.sample(n = 10)["Prompt"].map(lambda x: x).values.tolist()
@@ -219,13 +220,13 @@ with gr.Blocks(css=css) as demo:
219
  current_model = gr.Dropdown(label="Current Model", choices=list_models, value=list_models[1])
220
 
221
  with gr.Row(elem_id="prompt-container"):
222
- text_prompt = gr.Textbox(label="Prompt", placeholder="a cute dog", lines=1, elem_id="prompt-text-input", value = DEFAULT_PROMPT)
223
  text_button = gr.Button("Generate", variant='primary', elem_id="gen-button")
224
 
225
  with gr.Row("prompt-container"):
226
  with gr.Row():
227
  select_prompt = gr.Dropdown(label="Prompt selected", choices=list_prompts,
228
- value = "1girl, aqua eyes, baseball cap, blonde hair, closed mouth, earrings, green background, hat, hoop earrings, jewelry, looking at viewer, shirt, short hair, simple background, solo, upper body, yellow shirt",
229
  info = "default prompt: {}".format(DEFAULT_PROMPT)
230
  )
231
  select_button = gr.Button("Select Prompt Generate", variant='primary', elem_id="gen-button")
 
15
  prompt_df = pd.read_csv("Stable-Diffusion-Prompts.csv")
16
 
17
  #DEFAULT_MODEL = "stabilityai/stable-diffusion-2-1"
18
+ #DEFAULT_PROMPT = "1girl, aqua eyes, baseball cap, blonde hair, closed mouth, earrings, green background, hat, hoop earrings, jewelry, looking at viewer, shirt, short hair, simple background, solo, upper body, yellow shirt"
19
+ DEFAULT_PROMPT = "house"
20
 
21
  def get_samples():
22
  prompt_list = prompt_df.sample(n = 10)["Prompt"].map(lambda x: x).values.tolist()
 
220
  current_model = gr.Dropdown(label="Current Model", choices=list_models, value=list_models[1])
221
 
222
  with gr.Row(elem_id="prompt-container"):
223
+ text_prompt = gr.Textbox(label="Prompt", placeholder=DEFAULT_PROMPT, lines=1, elem_id="prompt-text-input", value = DEFAULT_PROMPT)
224
  text_button = gr.Button("Generate", variant='primary', elem_id="gen-button")
225
 
226
  with gr.Row("prompt-container"):
227
  with gr.Row():
228
  select_prompt = gr.Dropdown(label="Prompt selected", choices=list_prompts,
229
+ value = DEFAULT_PROMPT,
230
  info = "default prompt: {}".format(DEFAULT_PROMPT)
231
  )
232
  select_button = gr.Button("Select Prompt Generate", variant='primary', elem_id="gen-button")