Update app.py

app.py

@@ -16,7 +16,7 @@ MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
 @spaces.GPU
-def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
+def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, progress=gr.Progress(track_tqdm=True)):
 
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -55,8 +55,8 @@ with gr.Blocks(css=css) as demo:
     
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""
-        # Text-to-Image Gradio Template
-        Currently running on {power_device}.
+        # AuraFlow demo
+        Open source 5.6B parameters MMDiT model
         """)
         
         with gr.Row():
@@ -134,7 +134,8 @@ with gr.Blocks(css=css) as demo:
             cache_examples=False
         )
 
-    run_button.click(
+    gr.on(
+        triggers=[run_button.click, prompt.submit, negative_prompt.submit],
         fn = infer,
         inputs = [prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
         outputs = [result]