hcyyg commited on
Commit
379fb6c
1 Parent(s): adeed2e
Files changed (1) hide show
  1. 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": data.get("prompt"),
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