lachine commited on
Commit
d87d121
·
1 Parent(s): d4e62d8
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,10 +2,10 @@ from transformers import pipeline, set_seed
2
  from random import randint
3
  import gradio as gr
4
 
5
- generator = pipeline('text-generation', model='gpt2-large')
6
- set_seed(randint(randint(1000,10000),randint(50000,300000)))
7
 
8
  def gpt2(string, max_length):
 
9
  return generator(string, max_length=max_length, num_return_sequences=1)[0]['generated_text']
10
 
11
  max_length_slider = gr.inputs.Slider(minimum=50, maximum=500, step=10, default=100, label="Maximum Length")
 
2
  from random import randint
3
  import gradio as gr
4
 
5
+ generator = pipeline('text-generation', model='gpt2-medium')
 
6
 
7
  def gpt2(string, max_length):
8
+ set_seed(randint(randint(1000,10000),randint(50000,300000)))
9
  return generator(string, max_length=max_length, num_return_sequences=1)[0]['generated_text']
10
 
11
  max_length_slider = gr.inputs.Slider(minimum=50, maximum=500, step=10, default=100, label="Maximum Length")