Update edit_app.py

edit_app.py

@@ -70,15 +70,15 @@ def main():
             image_cfg_scale,
         )
 
-    def generate(
+     def generate(
         input_image: Image.Image,
         instruction: str,
-        steps: int,
-        randomize_seed: bool,
-        seed: int,
-        randomize_cfg: bool,
-        text_cfg_scale: float,
-        image_cfg_scale: float,
+        steps: int = 50,
+        randomize_seed: bool = True,
+        seed: int = 1371,
+        randomize_cfg: bool = True,
+        text_cfg_scale: float = 7.5,
+        image_cfg_scale: float = 1.5,
     ):
         seed = random.randint(0, 100000) if randomize_seed else seed
         text_cfg_scale = round(random.uniform(6.0, 9.0), ndigits=2) if randomize_cfg else text_cfg_scale
@@ -183,7 +183,21 @@ def main():
             inputs=[],
             outputs=[steps, randomize_seed, seed, randomize_cfg, text_cfg_scale, image_cfg_scale, edited_image],
         )
-
+        demo.add_endpoint(
+            fn=generate,
+            inputs=[
+                gr.Image(type="pil"),
+                gr.Textbox(),
+                gr.Number(default=50),
+                gr.Checkbox(default=True),
+                gr.Number(default=1371),
+                gr.Checkbox(default=True),
+                gr.Number(default=7.5),
+                gr.Number(default=1.5)
+            ],
+            outputs=[gr.Number(), gr.Number(), gr.Number(), gr.Image(type="pil")],
+            api_name="generate_image"
+    
     demo.queue(concurrency_count=1)
     demo.launch(share=False)