Update app.py

app.py

@@ -37,14 +37,14 @@ pipe_normal.scheduler = EDMEulerScheduler(sigma_min=0.002, sigma_max=120.0, sigm
 pipe_normal.to("cuda")
 
 @spaces.GPU
-def run_normal(prompt, progress=gr.Progress(track_tqdm=True)):
-    return pipe_normal(prompt, num_inference_steps=20).images[0]
+def run_normal(prompt, negative_prompt, guidance_scale, progress=gr.Progress(track_tqdm=True)):
+    return pipe_normal(prompt, negative_prompt=negative_prompt, guidance_scale=guidance_scale, num_inference_steps=20).images[0]
 
 @spaces.GPU
-def run_edit(image, prompt, progress=gr.Progress(track_tqdm=True)):
+def run_edit(image, prompt, negative_prompt, guidance_scale, progress=gr.Progress(track_tqdm=True)):
     resolution = 1024
     image.resize((resolution, resolution))
-    return pipe_edit(prompt=prompt,image=image,height=resolution,width=resolution,num_inference_steps=20).images[0]
+    return pipe_edit(prompt=prompt,image=image,height=resolution,width=resolution,negative_prompt=negative_prompt, guidance_scale=guidance_scale,num_inference_steps=20).images[0]
 css = '''
 .gradio-container{
 max-width: 768px !important;
@@ -62,7 +62,8 @@ with gr.Blocks(css=css) as demo:
             button_normal = gr.Button("Generate", min_width=120)
           output_normal = gr.Image(label="Your result image", interactive=False)
           with gr.Accordion("Advanced Settings", open=False):
-            pass
+            negative_prompt_normal = gr.Textbox(label="Negative Prompt")
+            guidance_scale_normal = gr.Number(label="Guidance Scale", value=7)
     with gr.Tab("CosXL Edit"):
       with gr.Group():
           image_edit = gr.Image(label="Image you would like to edit", type="pil")
@@ -71,7 +72,8 @@ with gr.Blocks(css=css) as demo:
             button_edit = gr.Button("Generate", min_width=120)
           output_edit = gr.Image(label="Your result image", interactive=False)
           with gr.Accordion("Advanced Settings", open=False):
-            pass
+            negative_prompt_edit = gr.Textbox(label="Negative Prompt")
+            guidance_scale_edit = gr.Number(label="Guidance Scale", value=7)
 
     button_edit.click(
         
@@ -82,7 +84,7 @@ with gr.Blocks(css=css) as demo:
             prompt_normal.submit
         ],
         fn=run_normal,
-        inputs=[prompt_normal],
+        inputs=[prompt_normal, negative_prompt_normal, guidance_scale_normal],
         outputs=[output_normal],
     )
     gr.on(
@@ -91,7 +93,7 @@ with gr.Blocks(css=css) as demo:
             prompt_edit.submit
         ],
         fn=run_edit,
-        inputs=[image_edit, prompt_edit],
+        inputs=[image_edit, prompt_edit, negative_prompt_edit, guidance_scale_edit],
         outputs=[output_edit]
     )
 if __name__ == "__main__":