csuvikv commited on
Commit
f4da77f
·
1 Parent(s): 6d0d2cb
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ generator = pipeline('text-generation', model='gpt2', return_full_text=False, do
8
 
9
 
10
  def Bemenet(input_string):
11
- return generator(os.environ['SECRET_PROMPT'] + " " + input_string, max_length=100, num_return_sequences=2)[1]['generated_text']
12
 
13
  textbox = gr.Textbox()
14
 
 
8
 
9
 
10
  def Bemenet(input_string):
11
+ return generator(os.environ['SECRET_PROMPT'] + " " + input_string, max_length=100)[0]['generated_text']
12
 
13
  textbox = gr.Textbox()
14