Spaces:
Running
on
Zero
Running
on
Zero
UI changes for advanced settings + keep presets available
Browse files
app.py
CHANGED
|
@@ -110,6 +110,17 @@ def infer(image, model_selection, width, height, overlap_width, num_inference_st
|
|
| 110 |
|
| 111 |
yield background, cnet_image
|
| 112 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 113 |
|
| 114 |
def clear_result():
|
| 115 |
return gr.update(value=None)
|
|
@@ -117,7 +128,7 @@ def clear_result():
|
|
| 117 |
|
| 118 |
css = """
|
| 119 |
.gradio-container {
|
| 120 |
-
width:
|
| 121 |
}
|
| 122 |
"""
|
| 123 |
|
|
@@ -138,41 +149,50 @@ with gr.Blocks(css=css) as demo:
|
|
| 138 |
sources=["upload"],
|
| 139 |
)
|
| 140 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 141 |
with gr.Row():
|
| 142 |
with gr.Column(scale=2):
|
| 143 |
prompt_input = gr.Textbox(label="Prompt (Optional)")
|
| 144 |
with gr.Column(scale=1):
|
| 145 |
run_button = gr.Button("Generate")
|
| 146 |
|
| 147 |
-
with gr.
|
| 148 |
-
|
| 149 |
-
|
| 150 |
-
|
| 151 |
-
|
| 152 |
-
|
| 153 |
-
|
| 154 |
-
|
| 155 |
-
|
| 156 |
-
|
| 157 |
-
|
| 158 |
-
|
| 159 |
-
|
| 160 |
-
|
| 161 |
-
|
| 162 |
-
|
| 163 |
-
|
| 164 |
-
|
| 165 |
-
|
| 166 |
-
|
| 167 |
-
|
| 168 |
-
|
| 169 |
-
|
| 170 |
-
|
| 171 |
-
|
| 172 |
-
|
| 173 |
-
|
| 174 |
-
|
| 175 |
-
|
|
|
|
|
|
|
|
|
|
| 176 |
|
| 177 |
gr.Examples(
|
| 178 |
examples=[
|
|
@@ -189,6 +209,12 @@ with gr.Blocks(css=css) as demo:
|
|
| 189 |
label="Generated Image",
|
| 190 |
)
|
| 191 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 192 |
run_button.click(
|
| 193 |
fn=clear_result,
|
| 194 |
inputs=None,
|
|
|
|
| 110 |
|
| 111 |
yield background, cnet_image
|
| 112 |
|
| 113 |
+
def preload_presets(target_ratio):
|
| 114 |
+
if target_ratio == "9:16":
|
| 115 |
+
changed_width = 720
|
| 116 |
+
changed_height = 1024
|
| 117 |
+
return changed_width, changed_height, gr.update(open=False)
|
| 118 |
+
elif target_ratio == "16:9":
|
| 119 |
+
changed_width = 1024
|
| 120 |
+
changed_height = 720
|
| 121 |
+
return changed_width, changed_height, gr.update(open=False)
|
| 122 |
+
elif target_ratio == "Custom":
|
| 123 |
+
return 720, 1024, gr.update(open=True)
|
| 124 |
|
| 125 |
def clear_result():
|
| 126 |
return gr.update(value=None)
|
|
|
|
| 128 |
|
| 129 |
css = """
|
| 130 |
.gradio-container {
|
| 131 |
+
width: 1800px !important;
|
| 132 |
}
|
| 133 |
"""
|
| 134 |
|
|
|
|
| 149 |
sources=["upload"],
|
| 150 |
)
|
| 151 |
|
| 152 |
+
target_ratio = gr.Radio(
|
| 153 |
+
label = "Expected Ratio",
|
| 154 |
+
choices = ["9:16", "16:9", "Custom"],
|
| 155 |
+
value = "9:16"
|
| 156 |
+
)
|
| 157 |
+
|
| 158 |
with gr.Row():
|
| 159 |
with gr.Column(scale=2):
|
| 160 |
prompt_input = gr.Textbox(label="Prompt (Optional)")
|
| 161 |
with gr.Column(scale=1):
|
| 162 |
run_button = gr.Button("Generate")
|
| 163 |
|
| 164 |
+
with gr.Accordion(label="Advanced settings", open=False) as settings_panel:
|
| 165 |
+
with gr.Column():
|
| 166 |
+
with gr.Row():
|
| 167 |
+
width_slider = gr.Slider(
|
| 168 |
+
label="Width",
|
| 169 |
+
minimum=720,
|
| 170 |
+
maximum=1440,
|
| 171 |
+
step=8,
|
| 172 |
+
value=720, # Set a default value
|
| 173 |
+
)
|
| 174 |
+
height_slider = gr.Slider(
|
| 175 |
+
label="Height",
|
| 176 |
+
minimum=720,
|
| 177 |
+
maximum=1440,
|
| 178 |
+
step=8,
|
| 179 |
+
value=1024, # Set a default value
|
| 180 |
+
)
|
| 181 |
+
with gr.Row():
|
| 182 |
+
model_selection = gr.Dropdown(
|
| 183 |
+
choices=list(MODELS.keys()),
|
| 184 |
+
value="RealVisXL V5.0 Lightning",
|
| 185 |
+
label="Model",
|
| 186 |
+
)
|
| 187 |
+
num_inference_steps = gr.Slider(label="Steps", minimum=4, maximum=12, step=1, value=8 )
|
| 188 |
+
|
| 189 |
+
overlap_width = gr.Slider(
|
| 190 |
+
label="Mask overlap width",
|
| 191 |
+
minimum=1,
|
| 192 |
+
maximum=50,
|
| 193 |
+
value=42,
|
| 194 |
+
step=1
|
| 195 |
+
)
|
| 196 |
|
| 197 |
gr.Examples(
|
| 198 |
examples=[
|
|
|
|
| 209 |
label="Generated Image",
|
| 210 |
)
|
| 211 |
|
| 212 |
+
target_ratio.change(
|
| 213 |
+
fn = preload_presets,
|
| 214 |
+
inputs = [target_ratio],
|
| 215 |
+
outputs = [width_slider, height_slider, settings_panel],
|
| 216 |
+
queue = False
|
| 217 |
+
)
|
| 218 |
run_button.click(
|
| 219 |
fn=clear_result,
|
| 220 |
inputs=None,
|