Update main.py

main.py

@@ -110,6 +110,6 @@ async def create_report(input: ReportInput):
     return {"report": report}
 
 @app.post("/get_images")
-async def get_images(input: ReportInput):
+async def get_images(input: TopicInput):
     images = get_images(input.user_input, input.num_topics)
     return {"images": images}