ehristoforu
commited on
Commit
·
51bd190
1
Parent(s):
b20c8c8
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ API_TOKEN = os.getenv("HF_READ_TOKEN") # it is free
|
|
10 |
headers = {"Authorization": f"Bearer {API_TOKEN}"}
|
11 |
|
12 |
|
13 |
-
def query(prompt, is_negative=False, image_style="None style", steps=
|
14 |
images = []
|
15 |
for _ in range(num_images):
|
16 |
if image_style == "None style":
|
@@ -196,10 +196,11 @@ with gr.Blocks(css=css) as demo:
|
|
196 |
"""
|
197 |
)
|
198 |
with gr.Group():
|
199 |
-
|
200 |
with gr.Box():
|
201 |
with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
|
202 |
with gr.Column():
|
|
|
203 |
text_prompt = gr.Textbox(show_label=False, placeholder="Enter your prompt", max_lines=1, elem_id="prompt-text-input").style(border=(True, False, True, True), rounded=(True, False, False, True), container=False)
|
204 |
negative_prompt = gr.Textbox(show_label=False, placeholder="Enter a negative prompt", max_lines=1, elem_id="negative-prompt-text-input").style(border=(True, False, True, True), rounded=(True, False, False, True), container=False)
|
205 |
text_button = gr.Button("Generate").style(margin=False, rounded=(False, True, True, False), full_width=False)
|
|
|
10 |
headers = {"Authorization": f"Bearer {API_TOKEN}"}
|
11 |
|
12 |
|
13 |
+
def query(prompt, is_negative=False, image_style="None style", steps=8, cfg_scale=7, seed=None, num_images=4):
|
14 |
images = []
|
15 |
for _ in range(num_images):
|
16 |
if image_style == "None style":
|
|
|
196 |
"""
|
197 |
)
|
198 |
with gr.Group():
|
199 |
+
|
200 |
with gr.Box():
|
201 |
with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
|
202 |
with gr.Column():
|
203 |
+
gallery_output = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
|
204 |
text_prompt = gr.Textbox(show_label=False, placeholder="Enter your prompt", max_lines=1, elem_id="prompt-text-input").style(border=(True, False, True, True), rounded=(True, False, False, True), container=False)
|
205 |
negative_prompt = gr.Textbox(show_label=False, placeholder="Enter a negative prompt", max_lines=1, elem_id="negative-prompt-text-input").style(border=(True, False, True, True), rounded=(True, False, False, True), container=False)
|
206 |
text_button = gr.Button("Generate").style(margin=False, rounded=(False, True, True, False), full_width=False)
|