Upload app.py

app.py

@@ -69,11 +69,11 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scal
     pipe.to("cuda")
     generator = torch.Generator(device="cuda").manual_seed(seed)
     
-    progress(0, desc="Start Inference.")
     with calculateDuration("Generating image"):
         # Generate image
         modes, images, scales = get_control_params()
         if not cn_on or controlnet is None or len(modes) == 0:
+            progress(0, desc="Start Inference.")
             image = pipe(
                 prompt=prompt_mash,
                 num_inference_steps=steps,
@@ -84,7 +84,8 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scal
                 joint_attention_kwargs={"scale": lora_scale},
             ).images[0]
         else:
-            
+            progress(0, desc="Start Inference with ControlNet.")
+            print(modes, scales) #
             image = pipe(
                 prompt=prompt_mash,
                 control_image=images,