Ashoka74 commited on
Commit
8521d3e
Β·
verified Β·
1 Parent(s): c3edff6

Update gradio_demo.py

Browse files
Files changed (1) hide show
  1. gradio_demo.py +1 -1
gradio_demo.py CHANGED
@@ -1174,7 +1174,7 @@ with block:
1174
  with gr.Tab("Background", visible=True):
1175
  # empty cache
1176
 
1177
- mask_mover = gr.State(MaskMover())
1178
  # with torch.no_grad():
1179
  # # Update the input channels to 12
1180
  # 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) # Changed from 8 to 12
 
1174
  with gr.Tab("Background", visible=True):
1175
  # empty cache
1176
 
1177
+ mask_mover = MaskMover()
1178
  # with torch.no_grad():
1179
  # # Update the input channels to 12
1180
  # 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) # Changed from 8 to 12