caobencheng commited on
Commit
4f258de
·
verified ·
1 Parent(s): c415d1c

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -0
main.py CHANGED
@@ -26,6 +26,8 @@ pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
26
 
27
  @app.get("/infer_t5")
28
  def infer_t5(input: str):
 
 
29
  if not input:
30
  return {"output": "Please enter some text."}
31
 
 
26
 
27
  @app.get("/infer_t5")
28
  def infer_t5(input: str):
29
+ # for test
30
+ return {"output": input}
31
  if not input:
32
  return {"output": "Please enter some text."}
33