Update app.py

app.py

@@ -133,7 +133,7 @@ def main():
                 pipe.to("cpu")
 
                 params = {'num_inference_steps': 100, 'num_images_per_prompt': 2}
-                generate_image(pipe, prompt, params)
+                generate_image(pipe, user_input, params)
                 
             else:
                 try: