ekinnk commited on
Commit
a191697
1 Parent(s): 7bbec8b

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -7,7 +7,7 @@ pipe_flan = pipeline("text-generation", model="gpt2")
7
 
8
  # Text generation
9
  def generator(input):
10
- output = pipe_flan(input, max_length=30, num_return_sequences=1)
11
  return output[0]["generated_text"]
12
 
13
  # Creating the Gradio Interface
 
7
 
8
  # Text generation
9
  def generator(input):
10
+ output = pipe_flan(input, max_length=50, num_return_sequences=1)
11
  return output[0]["generated_text"]
12
 
13
  # Creating the Gradio Interface