K1Z3M1112 commited on
Commit
2c95b32
·
verified ·
1 Parent(s): cdc06db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -4
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