InnoLink commited on
Commit
7d1c4ec
·
verified ·
1 Parent(s): 8564beb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -8,6 +8,9 @@ from transformers import pipeline
8
  app = FastAPI()
9
 
10
  pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
 
 
 
11
 
12
  @app.get("/infer_t5")
13
  def t5(input):
 
8
  app = FastAPI()
9
 
10
  pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
11
+ #Inference Augmentation: Load a dataset like SQuAD for question-answering;
12
+ #to generate responses, testing zero-shot capabilities
13
+
14
 
15
  @app.get("/infer_t5")
16
  def t5(input):