Update gradio_demo.py

gradio_demo.py

@@ -74,7 +74,7 @@ model.eval()
 # Change UNet
 
 with torch.no_grad():
-    new_conv_in = torch.nn.Conv2d(8, unet.conv_in.out_channels, unet.conv_in.kernel_size, unet.conv_in.stride, unet.conv_in.padding)
+    new_conv_in = torch.nn.Conv2d(12, unet.conv_in.out_channels, unet.conv_in.kernel_size, unet.conv_in.stride, unet.conv_in.padding)
     new_conv_in.weight.zero_()
     new_conv_in.weight[:, :4, :, :].copy_(unet.conv_in.weight)
     new_conv_in.bias = unet.conv_in.bias
@@ -129,12 +129,12 @@ unet.forward = hooked_unet_forward
 
 # Load
 
-model_path = './models/iclight_sd15_fc.safetensors'
-# model_path = './models/iclight_sd15_fbc.safetensors'
+#model_path = './models/iclight_sd15_fc.safetensors'
+model_path = './models/iclight_sd15_fbc.safetensors'
 
 
 if not os.path.exists(model_path):
-    download_url_to_file(url='https://huggingface.co/lllyasviel/ic-light/resolve/main/iclight_sd15_fc.safetensors', dst=model_path)
+    download_url_to_file(url='https://huggingface.co/lllyasviel/ic-light/resolve/main/iclight_sd15_fbc.safetensors', dst=model_path)
 
 sd_offset = sf.load_file(model_path)
 sd_origin = unet.state_dict()
@@ -763,7 +763,7 @@ def compress_image(image):
 
 block = gr.Blocks().queue()
 with block:
-    with gr.Tab("Text"):
+    with gr.Tab("Text", visible=False):
         with gr.Row():
             gr.Markdown("## Product Placement from Text")
         with gr.Row():
@@ -815,7 +815,7 @@ with block:
         example_quick_prompts.click(lambda x, y: ', '.join(y.split(', ')[:2] + [x[0]]), inputs=[example_quick_prompts, prompt], outputs=prompt, show_progress=False, queue=False)
         example_quick_subjects.click(lambda x: x[0], inputs=example_quick_subjects, outputs=prompt, show_progress=False, queue=False)
 
-    with gr.Tab("Background", visible=False):
+    with gr.Tab("Background", visible=True):
         mask_mover = MaskMover()