Update TextGen/router.py

TextGen/router.py

@@ -79,7 +79,7 @@ def api_home():
 
 @app.post("/api/generate", summary="Generate text from prompt", tags=["Generate"], response_model=Generate)
 def inference(message: Message):
-    return generate_text(prompt=message.messages)
+    return generate_text(messages=message.messages)
 
 #Dummy function for now
 def determine_vocie_from_npc(npc,genre):