Update app.py

app.py

@@ -97,9 +97,9 @@ with gr.Blocks() as demo:
                 cache_examples="lazy" 
             )
 
-    def enhance_image(*args):
+    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(*args))
+        return next(generate_image(prompt, current_seed, width, height))
 
     enhanceBtn.click(
         fn=enhance_image,