Spaces:
Build error
Build error
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")
|