small ui fix

src/app.py

@@ -79,15 +79,15 @@ with gr.Blocks(theme=theme) as demo:
                 with gr.Row():
                     controlnet_cond_scale = gr.Slider(
                         label="Controlnet Weight",
-                        value=0.5,
+                        value=1.0,
                         minimum=0.0,
-                        maximum=1.0,
+                        maximum=2.0,
                         step=0.1,
                     )
 
                 with gr.Row():
                     batch_size = gr.Slider(
-                        label="Batch Size", value=1, minimum=1, maximum=8, step=1
+                        label="Batch Size", value=1, minimum=1, maximum=4, step=1
                     )
                     seed = gr.Slider(-1, 2147483647, label="Seed", value=-1, step=1)
 
@@ -177,7 +177,7 @@ with gr.Blocks(theme=theme) as demo:
                     )
 
                 training_button = gr.Button(
-                    value="Train Style ControlNet", variant="primary"
+                    value="Train Anime ControlNet", variant="primary"
                 )
 
                 training_status = gr.Text(label="Training Status")

src/ui_functions.py

@@ -140,11 +140,11 @@ def generate(
 
     if seed == -1:
         seed = random.randint(0, 2147483647)
-
+    
     if guidance_image:
         guidance_image = extract_canny(guidance_image)
     else:
-        guidance_image = torch.zeros(n_images, 3, height, width)
+        guidance_image = torch.zeros(1, 3, height, width)
 
     generator = torch.Generator(device).manual_seed(seed)