Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
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 =
|
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")
|