Update app.py

app.py

@@ -97,12 +97,8 @@ with gr.Blocks() as demo:
                 cache_examples="lazy" 
             )
 
-    def enhance_image(prompt, current_seed, width, height):
-        gr.Info("Enhancing Image") # currently just runs optimized pipeline for 2 steps. Further implementations later.
-        return next(generate_image(prompt, current_seed, width, height))
-
     enhanceBtn.click(
-        fn=enhance_image,
+        fn=generate_image,
         inputs=[prompt, seed, width, height],
         outputs=[result, seed, latency],
         show_progress="full",