Rockramsri commited on
Commit
9fd1128
·
verified ·
1 Parent(s): eed9065
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -20,6 +20,7 @@ def home():
20
  @app.get("/generate")
21
  def generate(text:str):
22
  ## use the pipeline to generate text from given input text
 
23
  output== llm(text, max_tokens=1000)
24
  ## return the generate text in Json reposne
25
  return {"output":output["choices"][0]["text"]}
 
20
  @app.get("/generate")
21
  def generate(text:str):
22
  ## use the pipeline to generate text from given input text
23
+ print("Recieved prompt "+str(text))
24
  output== llm(text, max_tokens=1000)
25
  ## return the generate text in Json reposne
26
  return {"output":output["choices"][0]["text"]}