svjack commited on
Commit
7dcb97a
1 Parent(s): 9e256bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -40,7 +40,9 @@ prompt_df = dataset["train"].to_pandas()
40
  prompt_df = pd.read_csv("Stable-Diffusion-Prompts.csv")
41
 
42
  DEFAULT_MODEL = "stabilityai/stable-diffusion-2-1"
43
- 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"
 
 
44
 
45
  def get_samples():
46
  prompt_list = prompt_df.sample(n = 10)["Prompt"].map(lambda x: x).values.tolist()
@@ -279,13 +281,13 @@ with gr.Blocks(css=css) as demo:
279
  )
280
 
281
  with gr.Row("prompt-container"):
282
- text_prompt = gr.Textbox(label="Input Prompt", placeholder="a cute dog",
283
  value = DEFAULT_PROMPT,
284
  lines=2, elem_id="prompt-text-input")
285
  text_button = gr.Button("Manualy input Generate", variant='primary', elem_id="gen-button")
286
  with gr.Row("prompt-container"):
287
  select_prompt = gr.Dropdown(label="Prompt selected", choices=list_prompts,
288
- 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",
289
  info = "default prompt: {}".format(DEFAULT_PROMPT)
290
  )
291
  select_button = gr.Button("Select Prompt Generate", variant='primary', elem_id="gen-button")
 
40
  prompt_df = pd.read_csv("Stable-Diffusion-Prompts.csv")
41
 
42
  DEFAULT_MODEL = "stabilityai/stable-diffusion-2-1"
43
+ #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"
44
+ DEFAULT_PROMPT = "house"
45
+
46
 
47
  def get_samples():
48
  prompt_list = prompt_df.sample(n = 10)["Prompt"].map(lambda x: x).values.tolist()
 
281
  )
282
 
283
  with gr.Row("prompt-container"):
284
+ text_prompt = gr.Textbox(label="Input Prompt", placeholder=DEFAULT_PROMPT,
285
  value = DEFAULT_PROMPT,
286
  lines=2, elem_id="prompt-text-input")
287
  text_button = gr.Button("Manualy input Generate", variant='primary', elem_id="gen-button")
288
  with gr.Row("prompt-container"):
289
  select_prompt = gr.Dropdown(label="Prompt selected", choices=list_prompts,
290
+ value = DEFAULT_PROMPT,
291
  info = "default prompt: {}".format(DEFAULT_PROMPT)
292
  )
293
  select_button = gr.Button("Select Prompt Generate", variant='primary', elem_id="gen-button")