Update app.py

app.py

@@ -56,6 +56,8 @@ def main():
         instruction: str,
         seed: int,
         randomize_seed:bool,
+        text_cfg_scale: float,
+        image_cfg_scale: float,
     ):
         seed = random.randint(0, 100000) if randomize_seed else seed
         width, height = input_image.size
@@ -102,8 +104,8 @@ def main():
             edited_image = cv2.applyColorMap(edited_image, cv2.COLORMAP_JET)
             edited_image = Image.fromarray(cv2.cvtColor(edited_image, cv2.COLOR_BGR2RGB))
 
-        text_cfg_scale   = 7.5
-        image_cfg_scale  = 1.5
+        # text_cfg_scale   = 7.5
+        # image_cfg_scale  = 1.5
         return [seed, text_cfg_scale, image_cfg_scale, edited_image]
 
 
@@ -155,6 +157,8 @@ def main():
                 instruction,
                 seed,
                 randomize_seed,
+                text_cfg_scale,
+                image_cfg_scale,
             ],
             outputs=[seed, text_cfg_scale, image_cfg_scale, edited_image],
         )