SolomonHsu commited on
Commit
7026686
·
verified ·
1 Parent(s): 81e26a7

Update TextGen/router.py

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