rjiang12 commited on
Commit
bb57616
·
1 Parent(s): 6447e81

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,11 +4,11 @@ from transformers import pipeline
4
  generator = pipeline('text-generation', model='gpt2')
5
 
6
  def func(sentence, max_length, temp):
7
- output_list = generator(sentence, max_length=max_length, num_return_sequences=5, temperature=temp)
8
  output_strs = [dict['generated_text'] for dict in output_list]
9
  return output_strs
10
 
11
- demo = gr.Interface(fn=func, inputs=["text", gr.Slider(5, 50), gr.Slider(0.1, 100)], outputs=["text", "text", "text", "text", "text"])
12
 
13
  if __name__ == "__main__":
14
  demo.launch()
 
4
  generator = pipeline('text-generation', model='gpt2')
5
 
6
  def func(sentence, max_length, temp):
7
+ output_list = generator(sentence, max_length=max_length, num_return_sequences=5, temperature=float(temp))
8
  output_strs = [dict['generated_text'] for dict in output_list]
9
  return output_strs
10
 
11
+ demo = gr.Interface(fn=func, inputs=["text", gr.Slider(5, 25), gr.Slider(0.1, 100)], outputs=["text", "text", "text", "text", "text"])
12
 
13
  if __name__ == "__main__":
14
  demo.launch()