ianluo commited on
Commit
336e074
·
1 Parent(s): 2ce1aa8
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -5,7 +5,7 @@ from transformers import pipeline
5
 
6
  app = FastAPI()
7
 
8
- pipe_line = pipeline("text2text-generation", model="google/flan-t5-small")
9
 
10
  @app.get("/infer_t5")
11
  def t5(input):
 
5
 
6
  app = FastAPI()
7
 
8
+ pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
9
 
10
  @app.get("/infer_t5")
11
  def t5(input):