Update app.py

app.py

@@ -37,11 +37,11 @@ pipe_normal.scheduler = EDMEulerScheduler(sigma_min=0.002, sigma_max=120.0, sigm
 pipe_normal.to("cuda")
 
 @spaces.GPU
-def run_normal(prompt):
+def run_normal(prompt, progress=gr.Progress(track_tqdm=True)):
     return pipe_normal(prompt, num_inference_steps=20).images[0]
 
 @spaces.GPU
-def run_edit(image, prompt):
+def run_edit(image, prompt, 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]
@@ -51,7 +51,7 @@ max-width: 768px !important;
 margin: 0 auto;
 }
 '''
-with gr.Blocks() as demo:
+with gr.Blocks(css=css) as demo:
     gr.Markdown('''# CosXL demo
     Unofficial demo for CosXL, a SDXL model tuned to produce full color range images. CosXL Edit allows you to perform edits on images. Both have a [non-commercial community license](https://huggingface.co/stabilityai/cosxl/blob/main/LICENSE)
     ''')
@@ -72,12 +72,24 @@ with gr.Blocks() as demo:
           output_edit = gr.Image(label="Your result image", interactive=False)
           with gr.Accordion("Advanced Settings", open=False):
             pass
-    button_normal.click(
+
+    button_edit.click(
+        
+    )
+    gr.on(
+        triggers=[
+            button_normal.click,
+            prompt_normal.submit
+        ],
         fn=run_normal,
         inputs=[prompt_normal],
-        outputs=[output_normal]
+        outputs=[output_normal],
     )
-    button_edit.click(
+    gr.on(
+        triggers=[
+            button_edit.click,
+            prompt_edit.submit
+        ],
         fn=run_edit,
         inputs=[image_edit, prompt_edit],
         outputs=[output_edit]