Update TextGen/router.py

TextGen/router.py

@@ -59,7 +59,7 @@ def query(payload):
 @app.get("/query", tags=["Query"])
 def api_query(input: str):
     return {'input': input}
-    # return query({"inputs": input,"parameters":{"return_full_text":False,"max_length":2048}})
+    return query({"inputs": input,"parameters":{"return_full_text":False,"max_length":1024}})
 
 @app.post("/api/generate", summary="Generate text from prompt", tags=["Generate"], response_model=Generate)
 def inference(input_prompt: str):