Spaces:

osanseviero
/

i-like-flan

Running on A10G

osanseviero commited on Oct 22, 2022

Commit

4f3084f

•

1 Parent(s): a3210e4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ pipe = pipeline("text2text-generation", model="google/flan-t5-large")
 def inference(text):
   output = pipe(text)
-  return output["generated_text"]
 io = gr.Interface(inference, gr.Textbox(lines=3), gr.Textbox(lines=3))
 io.launch()

 def inference(text):
   output = pipe(text)
+  return output[0]["generated_text"]
 io = gr.Interface(inference, gr.Textbox(lines=3), gr.Textbox(lines=3))
 io.launch()