rjiang12 commited on
Commit
db5f61b
·
1 Parent(s): 940f979

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -1,9 +1,12 @@
1
  import gradio as gr
 
 
 
2
 
3
  def func(sentence):
4
- return f"your sentence was \"{sentence}\"!"
5
 
6
- demo = gr.Interface(fn=func, inputs="text", outputs="text")
7
 
8
  if __name__ == "__main__":
9
  demo.launch()
 
1
  import gradio as gr
2
+ from transformers import pipeline
3
+
4
+ generator = pipeline('text-generation', model='gpt2')
5
 
6
  def func(sentence):
7
+ return generator(sentence, max-length=30, num_return_sequences=5)
8
 
9
+ demo = gr.Interface(fn=func, inputs="text", outputs=["text", "text", "text", "text", "text"])
10
 
11
  if __name__ == "__main__":
12
  demo.launch()