Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -281,7 +281,7 @@ def enhance_image(
|
|
281 |
gr.Info(f"π Resizing output to target size: {w_original * upscale_factor}x{h_original * upscale_factor}")
|
282 |
image = image.resize((w_original * upscale_factor, h_original * upscale_factor), resample=Image.LANCZOS)
|
283 |
|
284 |
-
return [true_input_image, image]
|
285 |
|
286 |
|
287 |
# Create Gradio interface
|
@@ -386,20 +386,6 @@ with gr.Blocks(css=css, title="π¨ AI Image Upscaler - Florence-2 + FLUX") as d
|
|
386 |
height=600, # Made larger
|
387 |
elem_id="result_slider"
|
388 |
)
|
389 |
-
|
390 |
-
with gr.Row():
|
391 |
-
output_seed = gr.Number(
|
392 |
-
label="Used Seed",
|
393 |
-
precision=0,
|
394 |
-
interactive=False
|
395 |
-
)
|
396 |
-
|
397 |
-
generated_caption_output = gr.Textbox(
|
398 |
-
label="Generated Caption",
|
399 |
-
placeholder="AI-generated caption will appear here...",
|
400 |
-
lines=3,
|
401 |
-
interactive=False
|
402 |
-
)
|
403 |
|
404 |
# Event handler
|
405 |
enhance_btn.click(
|
@@ -415,20 +401,8 @@ with gr.Blocks(css=css, title="π¨ AI Image Upscaler - Florence-2 + FLUX") as d
|
|
415 |
use_generated_caption,
|
416 |
custom_prompt,
|
417 |
],
|
418 |
-
outputs=[result_slider
|
419 |
)
|
420 |
-
|
421 |
-
gr.HTML("""
|
422 |
-
<div style="margin-top: 2rem; padding: 1rem; background: #f0f0f0; border-radius: 8px;">
|
423 |
-
<h4>π‘ How it works:</h4>
|
424 |
-
<ol>
|
425 |
-
<li><strong>Florence-2</strong> analyzes your image and generates a detailed caption</li>
|
426 |
-
<li>Initial upscale with LANCZOS interpolation (or ESRGAN if installed)</li>
|
427 |
-
<li><strong>FLUX Img2Img</strong> upscales the upscaled image with tiled AI diffusion guided by the caption</li>
|
428 |
-
</ol>
|
429 |
-
<p><strong>Note:</strong> Output limited to 8192x8192 pixels total budget. Tiling enables larger sizes.</p>
|
430 |
-
</div>
|
431 |
-
""")
|
432 |
|
433 |
# Custom CSS for slider
|
434 |
gr.HTML("""
|
@@ -444,6 +418,19 @@ with gr.Blocks(css=css, title="π¨ AI Image Upscaler - Florence-2 + FLUX") as d
|
|
444 |
}
|
445 |
</style>
|
446 |
""")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
447 |
|
448 |
if __name__ == "__main__":
|
449 |
demo.queue().launch(share=True, server_name="0.0.0.0", server_port=7860)
|
|
|
281 |
gr.Info(f"π Resizing output to target size: {w_original * upscale_factor}x{h_original * upscale_factor}")
|
282 |
image = image.resize((w_original * upscale_factor, h_original * upscale_factor), resample=Image.LANCZOS)
|
283 |
|
284 |
+
return [true_input_image, image]
|
285 |
|
286 |
|
287 |
# Create Gradio interface
|
|
|
386 |
height=600, # Made larger
|
387 |
elem_id="result_slider"
|
388 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
389 |
|
390 |
# Event handler
|
391 |
enhance_btn.click(
|
|
|
401 |
use_generated_caption,
|
402 |
custom_prompt,
|
403 |
],
|
404 |
+
outputs=[result_slider]
|
405 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
406 |
|
407 |
# Custom CSS for slider
|
408 |
gr.HTML("""
|
|
|
418 |
}
|
419 |
</style>
|
420 |
""")
|
421 |
+
|
422 |
+
# JS to set slider default position to middle
|
423 |
+
gr.HTML("""
|
424 |
+
<script>
|
425 |
+
document.addEventListener('DOMContentLoaded', function() {
|
426 |
+
const sliderInput = document.querySelector('#result_slider input[type="range"]');
|
427 |
+
if (sliderInput) {
|
428 |
+
sliderInput.value = 50;
|
429 |
+
sliderInput.dispatchEvent(new Event('input'));
|
430 |
+
}
|
431 |
+
});
|
432 |
+
</script>
|
433 |
+
""")
|
434 |
|
435 |
if __name__ == "__main__":
|
436 |
demo.queue().launch(share=True, server_name="0.0.0.0", server_port=7860)
|