Spaces:
Runtime error
Runtime error
Update TextGen/router.py
Browse files- 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
|
| 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):
|