replace
Browse files- app/app.py +3 -2
app/app.py
CHANGED
@@ -26,10 +26,11 @@ app = FastAPI()
|
|
26 |
@app.post("/v1/engines/codegen/completions")
|
27 |
def engines(data=Body(None)):
|
28 |
print(data)
|
|
|
29 |
q = query({
|
30 |
-
"inputs":
|
31 |
})
|
32 |
-
return { "choices": [{"text":q[0].get("generated_text")}]}
|
33 |
def run():
|
34 |
uvicorn.run(app, host="0.0.0.0", port=7860)
|
35 |
|
|
|
26 |
@app.post("/v1/engines/codegen/completions")
|
27 |
def engines(data=Body(None)):
|
28 |
print(data)
|
29 |
+
prompt = data.get("prompt")
|
30 |
q = query({
|
31 |
+
"inputs": prompt,
|
32 |
})
|
33 |
+
return { "choices": [{"text":q[0].get("generated_text").replace(prompt,"",1)}]}
|
34 |
def run():
|
35 |
uvicorn.run(app, host="0.0.0.0", port=7860)
|
36 |
|