anyadecarlo commited on
Commit
bcf11ac
·
1 Parent(s): a9cbcd0
Files changed (1) hide show
  1. app.py +10 -10
app.py CHANGED
@@ -1,15 +1,15 @@
1
  import gradio as gr
2
- from transformers import pipeline
3
 
4
- pipe = pipeline("text2text-generation", model="google/flan-t5-small")
5
 
6
- def predict(prompt):
7
- return pipe(prompt)[0]["generated_text"]
8
 
9
- demo = gr.Interface(
10
- fn=predict,
11
- inputs="text",
12
- outputs="text"
13
- )
14
 
15
- demo.launch()
 
1
  import gradio as gr
2
+ from transformers import pipeline
3
 
4
+ pipe = pipeline("text2text-generation", model="google/flan-t5-small")
5
 
6
+ def predict(prompt):
7
+ return pipe(prompt)[0]["generated_text"]
8
 
9
+ demo = gr.Interface(
10
+ fn=predict,
11
+ inputs="text",
12
+ outputs="text"
13
+ )
14
 
15
+ demo.launch()