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