Spaces:
Paused
Paused
replace width\height sliders with dropdown
Browse files
app.py
CHANGED
|
@@ -88,15 +88,17 @@ def image_to_base64(image_path):
|
|
| 88 |
|
| 89 |
prodia_client = Prodia(api_key=os.getenv("PRODIA_API_KEY"))
|
| 90 |
|
| 91 |
-
def flip_text(prompt, negative_prompt, model, steps, sampler, cfg_scale,
|
|
|
|
|
|
|
| 92 |
|
| 93 |
config_without_model_and_sampler = {
|
| 94 |
"prompt": prompt,
|
| 95 |
"negative_prompt": negative_prompt,
|
| 96 |
"steps": steps,
|
| 97 |
"cfg_scale": cfg_scale,
|
| 98 |
-
"width": width,
|
| 99 |
-
"height": height,
|
| 100 |
"seed": seed
|
| 101 |
}
|
| 102 |
|
|
@@ -119,6 +121,17 @@ css = """
|
|
| 119 |
}
|
| 120 |
"""
|
| 121 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 122 |
with gr.Blocks(css=css) as demo:
|
| 123 |
|
| 124 |
|
|
@@ -149,9 +162,7 @@ with gr.Blocks(css=css) as demo:
|
|
| 149 |
|
| 150 |
with gr.Row():
|
| 151 |
with gr.Column(scale=1):
|
| 152 |
-
|
| 153 |
-
height = gr.Slider(label="Height", minimum=512, maximum=1536, value=1024, step=8)
|
| 154 |
-
gr.Markdown(elem_id="resolution", value="*Resolution Maximum: 1MP (1048576 px)*")
|
| 155 |
|
| 156 |
with gr.Column(scale=1):
|
| 157 |
batch_size = gr.Slider(label="Batch Size", maximum=1, value=1)
|
|
@@ -164,6 +175,6 @@ with gr.Blocks(css=css) as demo:
|
|
| 164 |
with gr.Column(scale=2):
|
| 165 |
image_output = gr.Image(value="https://cdn-uploads.huggingface.co/production/uploads/noauth/XWJyh9DhMGXrzyRJk7SfP.png")
|
| 166 |
|
| 167 |
-
text_button.click(flip_text, inputs=[prompt, negative_prompt, model, steps, sampler, cfg_scale,
|
| 168 |
|
| 169 |
demo.queue(default_concurrency_limit=10, max_size=32, api_open=False).launch(max_threads=128)
|
|
|
|
| 88 |
|
| 89 |
prodia_client = Prodia(api_key=os.getenv("PRODIA_API_KEY"))
|
| 90 |
|
| 91 |
+
def flip_text(prompt, negative_prompt, model, steps, sampler, cfg_scale, resolution, seed):
|
| 92 |
+
|
| 93 |
+
width, height = resolution.split("x")
|
| 94 |
|
| 95 |
config_without_model_and_sampler = {
|
| 96 |
"prompt": prompt,
|
| 97 |
"negative_prompt": negative_prompt,
|
| 98 |
"steps": steps,
|
| 99 |
"cfg_scale": cfg_scale,
|
| 100 |
+
"width": int(width),
|
| 101 |
+
"height": int(height),
|
| 102 |
"seed": seed
|
| 103 |
}
|
| 104 |
|
|
|
|
| 121 |
}
|
| 122 |
"""
|
| 123 |
|
| 124 |
+
list_resolutions = [
|
| 125 |
+
"1024x1024",
|
| 126 |
+
"1152x896",
|
| 127 |
+
"1216x832",
|
| 128 |
+
"1344x768",
|
| 129 |
+
"1536x640",
|
| 130 |
+
"640x1536",
|
| 131 |
+
"768x1344",
|
| 132 |
+
"832x1216"
|
| 133 |
+
]
|
| 134 |
+
|
| 135 |
with gr.Blocks(css=css) as demo:
|
| 136 |
|
| 137 |
|
|
|
|
| 162 |
|
| 163 |
with gr.Row():
|
| 164 |
with gr.Column(scale=1):
|
| 165 |
+
resolution = gr.Dropdown(value="1024x1024", show_label=True, label="Resolution", choices=list_resolutions)
|
|
|
|
|
|
|
| 166 |
|
| 167 |
with gr.Column(scale=1):
|
| 168 |
batch_size = gr.Slider(label="Batch Size", maximum=1, value=1)
|
|
|
|
| 175 |
with gr.Column(scale=2):
|
| 176 |
image_output = gr.Image(value="https://cdn-uploads.huggingface.co/production/uploads/noauth/XWJyh9DhMGXrzyRJk7SfP.png")
|
| 177 |
|
| 178 |
+
text_button.click(flip_text, inputs=[prompt, negative_prompt, model, steps, sampler, cfg_scale, resolution, seed], outputs=image_output)
|
| 179 |
|
| 180 |
demo.queue(default_concurrency_limit=10, max_size=32, api_open=False).launch(max_threads=128)
|