SolomonHsu commited on
Commit
45345d7
·
verified ·
1 Parent(s): 98d2562

Update TextGen/router.py

Browse files
Files changed (1) hide show
  1. TextGen/router.py +1 -1
TextGen/router.py CHANGED
@@ -57,7 +57,7 @@ def query(payload):
57
 
58
  @app.get("/query", tags=["Query"])
59
  def api_query(input: str):
60
- return query({"inputs": input},"parameters":{"return_full_text":False,"max_length":2048})
61
 
62
  @app.post("/api/generate", summary="Generate text from prompt", tags=["Generate"], response_model=Generate)
63
  def inference(input_prompt: str):
 
57
 
58
  @app.get("/query", tags=["Query"])
59
  def api_query(input: str):
60
+ return query({"inputs": input,"parameters":{"return_full_text":False,"max_length":2048}})
61
 
62
  @app.post("/api/generate", summary="Generate text from prompt", tags=["Generate"], response_model=Generate)
63
  def inference(input_prompt: str):