Update app.py
Browse files
app.py
CHANGED
|
@@ -48,10 +48,7 @@ CURRENT_T2I_PIPE = None
|
|
| 48 |
CURRENT_T2I_MODEL = None
|
| 49 |
|
| 50 |
# Define model types
|
| 51 |
-
SDXL_MODELS = [
|
| 52 |
-
"Laxhar/noobai-XL-1.1",
|
| 53 |
-
"madebyollin/sdxl-vae-fp16-fix" # เพิ่ม SDXL VAE fix model
|
| 54 |
-
]
|
| 55 |
SD15_MODELS = [
|
| 56 |
"digiplay/ChikMix_V3",
|
| 57 |
"digiplay/chilloutmix_NiPrunedFp16Fix",
|
|
@@ -536,6 +533,7 @@ def unload_all_models():
|
|
| 536 |
with gr.Blocks(title="🎨 Advanced Image Generation Suite", theme=gr.themes.Soft()) as demo:
|
| 537 |
gr.Markdown("# 🎨 Advanced Image Generation & Editing Suite")
|
| 538 |
gr.Markdown("### Powered by Stable Diffusion & ControlNet")
|
|
|
|
| 539 |
|
| 540 |
# Add system info
|
| 541 |
if torch.cuda.is_available():
|
|
@@ -552,6 +550,11 @@ with gr.Blocks(title="🎨 Advanced Image Generation Suite", theme=gr.themes.Sof
|
|
| 552 |
unload_btn.click(unload_all_models, outputs=status_text)
|
| 553 |
|
| 554 |
with gr.Tab("🎨 Colorize Sketch"):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 555 |
with gr.Row():
|
| 556 |
inp = gr.Image(label="Input Sketch/Image", type="pil")
|
| 557 |
out = gr.Image(label="Colored Output")
|
|
@@ -593,6 +596,12 @@ with gr.Blocks(title="🎨 Advanced Image Generation Suite", theme=gr.themes.Sof
|
|
| 593 |
)
|
| 594 |
|
| 595 |
with gr.Tab("🖼️ Text-to-Image"):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 596 |
with gr.Row():
|
| 597 |
t2i_out = gr.Image(label="Output", type="pil")
|
| 598 |
|
|
|
|
| 48 |
CURRENT_T2I_MODEL = None
|
| 49 |
|
| 50 |
# Define model types
|
| 51 |
+
SDXL_MODELS = ["Laxhar/noobai-XL-1.1"] # เพิ่ม SDXL models ตรงนี้
|
|
|
|
|
|
|
|
|
|
| 52 |
SD15_MODELS = [
|
| 53 |
"digiplay/ChikMix_V3",
|
| 54 |
"digiplay/chilloutmix_NiPrunedFp16Fix",
|
|
|
|
| 533 |
with gr.Blocks(title="🎨 Advanced Image Generation Suite", theme=gr.themes.Soft()) as demo:
|
| 534 |
gr.Markdown("# 🎨 Advanced Image Generation & Editing Suite")
|
| 535 |
gr.Markdown("### Powered by Stable Diffusion & ControlNet")
|
| 536 |
+
gr.Markdown("**Note:** SDXL models (noobai-XL) work only in Text-to-Image tab, not in ControlNet")
|
| 537 |
|
| 538 |
# Add system info
|
| 539 |
if torch.cuda.is_available():
|
|
|
|
| 550 |
unload_btn.click(unload_all_models, outputs=status_text)
|
| 551 |
|
| 552 |
with gr.Tab("🎨 Colorize Sketch"):
|
| 553 |
+
gr.Markdown("""
|
| 554 |
+
### Convert your sketches to colored images using ControlNet
|
| 555 |
+
**Note:** SDXL models are not compatible with ControlNet. Please use SD1.5 models only.
|
| 556 |
+
""")
|
| 557 |
+
|
| 558 |
with gr.Row():
|
| 559 |
inp = gr.Image(label="Input Sketch/Image", type="pil")
|
| 560 |
out = gr.Image(label="Colored Output")
|
|
|
|
| 596 |
)
|
| 597 |
|
| 598 |
with gr.Tab("🖼️ Text-to-Image"):
|
| 599 |
+
gr.Markdown("""
|
| 600 |
+
### Generate images from text descriptions
|
| 601 |
+
Supports both SD1.5 and SDXL models.
|
| 602 |
+
**Tip:** SDXL models produce higher quality but require more memory.
|
| 603 |
+
""")
|
| 604 |
+
|
| 605 |
with gr.Row():
|
| 606 |
t2i_out = gr.Image(label="Output", type="pil")
|
| 607 |
|