Update TextGen/router.py

TextGen/router.py

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