Update app.py

app.py

@@ -150,7 +150,8 @@ def generate(
         "add_quality_tags": add_quality_tags,
         "quality_tags": quality_selector,
         "isImg2Img": isImg2Img,
-        "img_path": img_path
+        "img_path": img_path,
+        "img2img_strength": img2img_strength
     }
 
     if use_upscaler:
@@ -174,6 +175,7 @@ def generate(
     try:
         if use_upscaler:
             if isImg2Img:
+                print("Img2Img")
                 img, img_width, img_height = load_img(512, img_path)
                 latents = img_pipe(
                     prompt=prompt,
@@ -222,8 +224,9 @@ def generate(
                 ).images
         else:
             if isImg2Img:
+                print("Img2Img")
                 img, img_width, img_height = load_img(512, img_path)
-                images = pipe(
+                images = img_pipe(
                     prompt=prompt,
                     negative_prompt=negative_prompt,
                     width=img_width,
@@ -236,7 +239,7 @@ def generate(
                     output_type="pil",
                 ).images
             else:
-                images = img_pipe(
+                images = pipe(
                     prompt=prompt,
                     negative_prompt=negative_prompt,
                     width=width,