avreymi commited on
Commit
10631e7
1 Parent(s): 372c8e4
Files changed (1) hide show
  1. t5.py +1 -1
t5.py CHANGED
@@ -5,4 +5,4 @@ pipe_flan = pipeline(model="bigscience/bloom-560m")
5
 
6
  def t5(input):
7
  output = pipe_flan(input)
8
- return output
 
5
 
6
  def t5(input):
7
  output = pipe_flan(input)
8
+ return output[0]["generated_text"]