Spaces:
Runtime error
Runtime error
Update gradio_demo.py
Browse files- 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 =
|
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
|