Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -191,6 +191,43 @@ DEFAULT_VALUES = {
|
|
| 191 |
"color_fix_method": "adain"
|
| 192 |
}
|
| 193 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 194 |
# Create interface components
|
| 195 |
with gr.Blocks(title="Controllable Conditional Super-Resolution") as demo:
|
| 196 |
gr.Markdown("## Controllable Conditional Super-Resolution")
|
|
@@ -200,7 +237,6 @@ with gr.Blocks(title="Controllable Conditional Super-Resolution") as demo:
|
|
| 200 |
with gr.Column():
|
| 201 |
input_image = gr.Image(label="Input Image")
|
| 202 |
|
| 203 |
-
# Put all parameters in an accordion/dropdown
|
| 204 |
with gr.Accordion("Advanced Options", open=False):
|
| 205 |
prompt = gr.Textbox(label="Prompt", value=DEFAULT_VALUES["prompt"])
|
| 206 |
negative_prompt = gr.Textbox(label="Negative Prompt", value=DEFAULT_VALUES["negative_prompt"])
|
|
@@ -215,7 +251,6 @@ with gr.Blocks(title="Controllable Conditional Super-Resolution") as demo:
|
|
| 215 |
value=DEFAULT_VALUES["color_fix_method"]
|
| 216 |
)
|
| 217 |
|
| 218 |
-
# Add buttons
|
| 219 |
with gr.Row():
|
| 220 |
clear_btn = gr.Button("Clear")
|
| 221 |
submit_btn = gr.Button("Submit", variant="primary")
|
|
@@ -223,6 +258,19 @@ with gr.Blocks(title="Controllable Conditional Super-Resolution") as demo:
|
|
| 223 |
with gr.Column():
|
| 224 |
output_image = gr.Image(label="Generated Image")
|
| 225 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 226 |
# Define submit action
|
| 227 |
submit_btn.click(
|
| 228 |
fn=process_image,
|
|
|
|
| 191 |
"color_fix_method": "adain"
|
| 192 |
}
|
| 193 |
|
| 194 |
+
# Define example data
|
| 195 |
+
EXAMPLES = [
|
| 196 |
+
[
|
| 197 |
+
"examples/low_res_1.jpg", # Input image path
|
| 198 |
+
"clean, detailed photo, high quality, 8k resolution", # Prompt
|
| 199 |
+
"blurry, noise, artifacts, low quality", # Negative prompt
|
| 200 |
+
3.0, # Guidance scale
|
| 201 |
+
1.0, # Conditioning scale
|
| 202 |
+
6, # Num steps
|
| 203 |
+
42, # Seed
|
| 204 |
+
4, # Upscale factor
|
| 205 |
+
"wavelet" # Color fix method
|
| 206 |
+
],
|
| 207 |
+
[
|
| 208 |
+
"examples/low_res_2.jpg",
|
| 209 |
+
"sharp, crisp details, professional photo, ultra HD",
|
| 210 |
+
"blur, pixelated, low resolution, noise",
|
| 211 |
+
3.0,
|
| 212 |
+
1.0,
|
| 213 |
+
6,
|
| 214 |
+
123,
|
| 215 |
+
4,
|
| 216 |
+
"wavelet"
|
| 217 |
+
],
|
| 218 |
+
[
|
| 219 |
+
"examples/low_res_3.jpg",
|
| 220 |
+
"natural texture, fine details, pristine quality",
|
| 221 |
+
"artificial, smooth, blurry, low quality",
|
| 222 |
+
2.5,
|
| 223 |
+
1.0,
|
| 224 |
+
6,
|
| 225 |
+
456,
|
| 226 |
+
4,
|
| 227 |
+
"wavelet"
|
| 228 |
+
]
|
| 229 |
+
]
|
| 230 |
+
|
| 231 |
# Create interface components
|
| 232 |
with gr.Blocks(title="Controllable Conditional Super-Resolution") as demo:
|
| 233 |
gr.Markdown("## Controllable Conditional Super-Resolution")
|
|
|
|
| 237 |
with gr.Column():
|
| 238 |
input_image = gr.Image(label="Input Image")
|
| 239 |
|
|
|
|
| 240 |
with gr.Accordion("Advanced Options", open=False):
|
| 241 |
prompt = gr.Textbox(label="Prompt", value=DEFAULT_VALUES["prompt"])
|
| 242 |
negative_prompt = gr.Textbox(label="Negative Prompt", value=DEFAULT_VALUES["negative_prompt"])
|
|
|
|
| 251 |
value=DEFAULT_VALUES["color_fix_method"]
|
| 252 |
)
|
| 253 |
|
|
|
|
| 254 |
with gr.Row():
|
| 255 |
clear_btn = gr.Button("Clear")
|
| 256 |
submit_btn = gr.Button("Submit", variant="primary")
|
|
|
|
| 258 |
with gr.Column():
|
| 259 |
output_image = gr.Image(label="Generated Image")
|
| 260 |
|
| 261 |
+
# Add examples
|
| 262 |
+
gr.Examples(
|
| 263 |
+
examples=EXAMPLES,
|
| 264 |
+
inputs=[
|
| 265 |
+
input_image, prompt, negative_prompt, guidance_scale,
|
| 266 |
+
conditioning_scale, num_steps, seed, upscale_factor,
|
| 267 |
+
color_fix_method
|
| 268 |
+
],
|
| 269 |
+
outputs=output_image,
|
| 270 |
+
fn=process_image,
|
| 271 |
+
cache_examples=True # Cache the results for faster loading
|
| 272 |
+
)
|
| 273 |
+
|
| 274 |
# Define submit action
|
| 275 |
submit_btn.click(
|
| 276 |
fn=process_image,
|