Update gradio_demo.py

gradio_demo.py

@@ -86,7 +86,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
@@ -142,8 +142,8 @@ unet.forward = hooked_unet_forward
 # Load
 
 # Model paths
-model_path = './models/iclight_sd15_fc.safetensors'
-model_path2 = './checkpoints/depth_anything_v2_vits.pth'
+model_path = './models/iclight_sd15_fbc.safetensors'
+model_path = './checkpoints/depth_anything_v2_vits.pth'
 model_path3 = './checkpoints/sam2_hiera_large.pt'
 model_path4 = './checkpoints/config.json'
 model_path5 = './checkpoints/preprocessor_config.json'
@@ -155,7 +155,7 @@ BASE_URL = 'https://huggingface.co/Ashoka74/Placement/resolve/main/'
 
 # Model URLs
 model_urls = {
-    model_path: 'iclight_sd15_fc.safetensors',
+    model_path: 'iclight_sd15_fbc.safetensors',
     model_path2: 'depth_anything_v2_vits.pth',
     model_path3: 'sam2_hiera_large.pt',
     model_path4: 'config.json',
@@ -186,8 +186,8 @@ ensure_directories()
 download_models()
 
 
-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)
+# 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)
 
 sd_offset = sf.load_file(model_path)
 sd_origin = unet.state_dict()
@@ -953,7 +953,7 @@ def process_image(input_image, input_text):
 
 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():
@@ -1044,7 +1044,7 @@ with block:
             outputs=[extracted_objects, extracted_fg], show_progress=True
             )
         
-    with gr.Tab("Background", visible=False):
+    with gr.Tab("Background", visible=True):
         # empty cache
     
         mask_mover = MaskMover()